Same hyper-parameters as original script

RubixML · Mar 31, 2023 · 1bcb7f8 · 1bcb7f8
1 parent d510b75
commit 1bcb7f8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
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([