schema: '2.0'
stages:
  train:
    cmd: poetry run train
    deps:
    - path: data/raw/arxivData.json
      md5: a314e2f4eab544a46e6f95802ecde647
      size: 72422946
    - path: shad_mlops_transformers/model.py
      md5: 465d64e91ac7d713e57442c932a665ea
      size: 2889
    - path: shad_mlops_transformers/trainer.py
      md5: 39d02f490890ef306b1165fcc29b8895
      size: 7495
    params:
      shad_mlops_transformers/config.py:
        Config.batch_size: 64
        Config.epochs: 15
        Config.random_seed: 42
        Config.test_size: 0.2
    outs:
    - path: data/models/model.torch
      md5: cad526c49e9a9ef312e820ba4cc93442
      size: 438842393
    - path: data/plots/train.png
      md5: a23fefe43ee875860f875123bd402522
      size: 40498
    - path: data/raw/mapper.json
      md5: e4c22adf587676d22ed58c812f41f409
      size: 855