diff --git a/NoSoliciting.Trainer/Program.cs b/NoSoliciting.Trainer/Program.cs index 8e89aa6..9d0e456 100644 --- a/NoSoliciting.Trainer/Program.cs +++ b/NoSoliciting.Trainer/Program.cs @@ -95,7 +95,9 @@ namespace NoSoliciting.Trainer { var model = pipeline.Fit(train); - ctx.Model.Save(model, train.Schema, @"../../../model.zip"); + if (full) { + ctx.Model.Save(model, train.Schema, @"../../../model.zip"); + } var testPredictions = model.Transform(ttd.TestSet); var eval = ctx.MulticlassClassification.Evaluate(testPredictions);