Hi!
I am trying to replicate your work. Repository does not contain feature extractor model nor model for transfer learning so I changed in settings.py:
(line 139) sel_model = Model.CLASSIFIER_MCDCNN
(line 143) sel_dataset = Dataset.BALABIT
(line 177) sel_occ_features = OCCFeatures.RAW_X_Y_DIR
but now I get dimension error ("expected input_layer to have 3 dimensions, but got array with shape (230, 256)").
How can I train PLAIN model?
Thanks!