Skip to content

Pre-trained models at paper submission

Latest
Compare
Choose a tag to compare
@malteos malteos released this 09 Aug 11:24
· 3 commits to master since this release

Each pre-trained model consists of the following files:

  • best_thresholds.csv: Best classification thresholds for each label (not available for final models)
  • model_config.json: Model hyper-parameter, e.g. hidden layer size.
  • model_weights: Trained model weights
  • outputs_with_ids.pickle: Prediction for validation set
  • report.json/txt: Classification report (output of sklearn.metrics.classification_report, not available for final models)
  • report.json/txt: Classification report only for books with available author embeddings (output of sklearn.metrics.classification_report, not available for final models)
  • submission.txt: Prediction for validation/test set in output format for shared task.