Update params.yaml:featurize.max_features, pa...

...rams.yaml:featurize.ngrams, params.yaml:prepare.split, params.yaml:train.min_split, params.yaml:train.n_est
shivpalSW · Jul 19, 2023 · dd297cb · dd297cb
1 parent 085373a
commit dd297cb
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/params.yaml b/params.yaml
@@ -1,14 +1,14 @@
 # This contains params to be used by the stages to train or predict
 prepare:
-  split: 0.4
+  split: 0.2
   seed: 2021
 
 featurize:
-  max_features: 1000
-  ngrams: 4
+  max_features: 500
+  ngrams: 2
 
 
 train:
   seed: 2021
-  n_est: 50
-  min_split: 8
+  n_est: 100
+  min_split: 6