| { | |
| "name": "TinyModel1", | |
| "version": "0.3.0", | |
| "task": "text-classification", | |
| "dataset": "fancyzhx/ag_news", | |
| "dataset_config": null, | |
| "train_split": "train", | |
| "eval_split": "test", | |
| "text_column": "text", | |
| "label_column": "label", | |
| "base_model": "tinymodel1-bert-scratch", | |
| "labels": [ | |
| "World", | |
| "Sports", | |
| "Business", | |
| "Sci/Tech" | |
| ], | |
| "eval_accuracy": 0.5383, | |
| "eval_macro_f1": 0.4554, | |
| "eval_weighted_f1": 0.4527, | |
| "eval_per_class_f1": { | |
| "World": 0.5366, | |
| "Sports": 0.731, | |
| "Business": 0.0, | |
| "Sci/Tech": 0.5539 | |
| }, | |
| "train_loss": 1.1567, | |
| "num_parameters": 1339268, | |
| "max_train_samples": 3000, | |
| "max_eval_samples": 600, | |
| "max_seq_length": 128, | |
| "epochs": 2, | |
| "batch_size": 16, | |
| "learning_rate": 0.0001, | |
| "seed": 42 | |
| } | |