{
  "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
}