Rishav-hub
/
dvc-nlp-demo
mirror of https://github.com/Rishav-hub/dvc-nlp-demo.git


  
1

	
2

	
3

	
4

	
5

	
6

	
7

	
8

	
9

	
10

	
11

	
12

	
13

	
            # This contains params to be used by the stages to train or predict
prepare:
  split: 0.2
  seed: 2021

featurize:
  ngrams: 2
  max_features: 3000

train:
  seed: 2021
  n_est: 70 
  min_split: 8