chore(trainer): only save model on full run
This commit is contained in:
parent
0a575866fa
commit
09995d3cf9
|
@ -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);
|
||||
|
|
Loading…
Reference in New Issue