From 2f7761b9b0addc9c74412aca78b49c4465dd440c Mon Sep 17 00:00:00 2001
From: Anna <git@anna.lgbt>
Date: Sat, 2 Jan 2021 07:31:34 -0500
Subject: [PATCH] chore(trainer): only save model on full run

---
 NoSoliciting.Trainer/Program.cs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

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);