Skip to content

[train_task_classifyapp] doesn't have an opportunity to use pre-trained model weights  #7

@natalymr

Description

@natalymr

I've noticed that there was missing a method in NCC_classifyapp class:

def load_weights(self, file_with_weights):
        self.model.load_weights(file_with_weights)

So, with the aim to just load weights and test the model in evaluate method you can do smth like this:

model.load_weights('published_results/classifyapp/CLASSIFYAPP-94.83.h5')

# Test model
print('\n--- Testing model...')
p = model.predict_gen(generator=gen_test)[0]

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions