diff --git a/params.yaml b/params.yaml index 9f5d7dd..fe030a2 100644 --- 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.2 + split: 0.3 seed: 2021 featurize: - max_features: 500 - ngrams: 2 + max_features: 3000 + ngrams: 3 train: seed: 2021 - n_est: 100 - min_split: 6 + n_est: 90 + min_split: 16