Merge pull request #2 from shivpalSW/main-30e56dd

Update params.yaml:featurize.max_features, pa...
shivpalSW · Jul 18, 2023 · 31c152c · 31c152c
2 parents 30e56dd + 1330d72
commit 31c152c
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