Reverted to old features

smith-chem-wisc · Sep 4, 2024 · ecd3ef8 · ecd3ef8
1 parent 979119f
commit ecd3ef8
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/MetaMorpheus/EngineLayer/FdrAnalysis/PEPAnalysisEngine.cs b/MetaMorpheus/EngineLayer/FdrAnalysis/PEPAnalysisEngine.cs
@@ -34,14 +34,13 @@ public Microsoft.ML.Trainers.FastTree.FastTreeBinaryTrainer.Options GetFastTreeO
                 NumberOfThreads = maxThreads,
                 NumberOfTrees = 400, 
                 MinimumExampleCountPerLeaf = 10,
-                MaximumBinCountPerFeature = 255,
+                NumberOfLeaves = 20,
                 LearningRate = 0.2,
-                Shrinkage = 0.1,
                 LabelColumnName = "Label",
                 FeatureColumnName = "Features",
-                UnbalancedSets = true,
                 Seed = 42,
-                FeatureSelectionSeed = 42
+                FeatureSelectionSeed = 42,
+                UnbalancedSets = false // You might think this should be set to true, but you'd be wrong
             };
         }