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 18, 2023 · 1330d72 · 1330d72
1 parent 30e56dd
commit 1330d72
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.3
+  split: 0.2
   seed: 2021
 
 featurize:
-  max_features: 2000
-  ngrams: 3
+  max_features: 3000
+  ngrams: 2
 
 
 train:
   seed: 2021
-  n_est: 100
-  min_split: 16
+  n_est: 80
+  min_split: 8