Skip to content

Load class weights for non-CNN models #235

@ianbulovic

Description

@ianbulovic

When I was refactoring for #230, I noticed that the class weights were only ever loaded for the CNN model. For the Hierarchical and CNLP models the class weights were always taken from the legacy dataset.class_weights which (as far as I can tell) was always None. To avoid changing functionality in the refactor I left these as None in the new code, but this is almost certainly a bug.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions