File size: 772 Bytes
dbde804 6df69f6 dbde804 6df69f6 dbde804 6df69f6 0ba0739 6df69f6 0ba0739 6df69f6 dbde804 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 | {
"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.515,
"eval_macro_f1": 0.4262,
"eval_weighted_f1": 0.424,
"eval_per_class_f1": {
"World": 0.4268,
"Sports": 0.7114,
"Business": 0.0,
"Sci/Tech": 0.5665
},
"train_loss": 1.1535,
"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
}
|