Same hyper-parameters as original script

andrewdalpino · andrewdalpino · commit 1bcb7f819d36 · 2023-03-31T12:31:25.000-05:00
diff --git a/README.md b/README.md
@@ -84,7 +84,7 @@ use Rubix\ML\Persisters\Filesystem;
 $estimator = new PersistentModel(
     new Pipeline([
         new TextNormalizer(),
-        new WordCountVectorizer(10000, 0.00008, 0.4, new NGram(1, 2)),
+        new WordCountVectorizer(10000, 2, 0.4, new NGram(1, 2)),
         new TfIdfTransformer(),
         new ZScaleStandardizer(),
     ], new MultilayerPerceptron([
diff --git a/train.php b/train.php
@@ -41,7 +41,7 @@
 $estimator = new PersistentModel(
     new Pipeline([
         new TextNormalizer(),
-        new WordCountVectorizer(10000, 5, 0.4, new NGram(1, 2)),
+        new WordCountVectorizer(10000, 2, 0.4, new NGram(1, 2)),
         new TfIdfTransformer(),
         new ZScaleStandardizer(),
     ], new MultilayerPerceptron([
