add key to class plot to denote xgb vs dnn
add key to class plot to denote xgb vs dnn