Releases: malteos/pytorch-bert-document-classification
Releases · malteos/pytorch-bert-document-classification
Pre-trained models at paper submission
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 weightsoutputs_with_ids.pickle: Prediction for validation setreport.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.