arjvik
/
dvc-demo
mirror of https://github.com/arjvik/dvc-demo


  
1

	
2

	
3

	
4

	
5

	
6

	
7

	
8

	
9

	
10

	
11

	
12

	
13

	
14

	
15

	
16

	
17

	
18

	
19

	
20

	
21

	
22

	
            paths:
  dataset: 'data/CrossValidated-Questions-Nov-2020.csv'
  train_df: 'outputs/train.csv.gz'
  test_df: 'outputs/test.csv.gz'
  train_df_featurized: 'outputs/train-featurized.csv.gz'
  test_df_featurized: 'outputs/test-featurized.csv.gz'
  train_tfidf: 'outputs/train-tfidf.npz'
  test_tfidf: 'outputs/test-tfidf.npz'
  tfidf: 'outputs/tfidf.pkl'
  model: 'outputs/model.pkl'
  train_metrics: 'metrics-train.yaml'
  test_metrics: 'metrics-test.yaml'

split:
  seed: 42

tfidf:
  max_features: 25000

train:
  loss: 'modified_huber'
  seed: 42