dvc.lock

schema: '2.0'
stages:
  process:
    cmd: python src/process_data.py
    deps:
    - path: data/raw
      md5: 84a0e37242f885ea418b9953761d35de.dir
      size: 84199
      nfiles: 2
    - path: src/process_data.py
      md5: 996ccb3e0f024a915417f77a58ad9cde
      size: 1237
    params:
      params.yaml:
        data:
          raw: data/raw/winequality-red.csv
          intermediate: data/intermediate
        process:
          feature: quality
          test_size: 0.2
    outs:
    - path: data/intermediate
      md5: 3377ebd11434a04b64fe3ca5cb3cc455.dir
      size: 194875
      nfiles: 4
  train:
    cmd: python src/train.py
    deps:
    - path: data/intermediate
      md5: 3377ebd11434a04b64fe3ca5cb3cc455.dir
      size: 194875
      nfiles: 4
    - path: src/train.py
      md5: 9c7de68b1bdc5dae97e849818a7bc1d2
      size: 1652
    params:
      params.yaml:
        data:
          raw: data/raw/winequality-red.csv
          intermediate: data/intermediate
        model: model/svm
        train:
          hyperparameters:
            svm__kernel:
            - linear
            - rbf
            svm__C:
            - 0.1
            - 1
            - 10
            svm__gamma:
            - 0.1
            - 1
            - 10
          grid_search:
            cv: 2
            scoring: accuracy
            verbose: 3
    outs:
    - path: model/svm
      md5: 56a719e06202368c953540a7cb3e37ef
      size: 142616
    - path: model/svm.mlem
      md5: b562c8482a0cb77f10bf8b14bf65f548
      size: 1263
  evaluate:
    cmd: python src/evaluate.py
    deps:
    - path: data/intermediate
      md5: 3377ebd11434a04b64fe3ca5cb3cc455.dir
      size: 194875
      nfiles: 4
    - path: model/svm
      md5: 56a719e06202368c953540a7cb3e37ef
      size: 142616
    - path: src/evaluate.py
      md5: 912aac9c6e00b70858f1d47bd16e51a0
      size: 803
    params:
      params.yaml:
        data:
          raw: data/raw/winequality-red.csv
          intermediate: data/intermediate
        model: model/svm
    outs:
    - path: dvclive/metrics.json
      md5: 62b3730eec31d2778145d46759a8a45f
      size: 27