Skip to content

Running training script without feature extractor model #1

@KamilaLis

Description

@KamilaLis

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!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions