Skip to content

Adapt the neural network for multi-class quality prediction #1

Description

@Thunder-Blaze

Transition the target from quality_binary to the full multi-class quality score. Update the output layer nodes, change the loss function to CrossEntropy, and tune the architecture (e.g., adding a second hidden layer or dropout) to maximize the multi-class F1-score.

Metadata

Metadata

Assignees

No one assigned

    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