chore(trainer): only save model on full run
This commit is contained in:
parent
a07f21ddd3
commit
2f7761b9b0
@ -95,7 +95,9 @@ namespace NoSoliciting.Trainer {
|
|||||||
|
|
||||||
var model = pipeline.Fit(train);
|
var model = pipeline.Fit(train);
|
||||||
|
|
||||||
|
if (full) {
|
||||||
ctx.Model.Save(model, train.Schema, @"../../../model.zip");
|
ctx.Model.Save(model, train.Schema, @"../../../model.zip");
|
||||||
|
}
|
||||||
|
|
||||||
var testPredictions = model.Transform(ttd.TestSet);
|
var testPredictions = model.Transform(ttd.TestSet);
|
||||||
var eval = ctx.MulticlassClassification.Evaluate(testPredictions);
|
var eval = ctx.MulticlassClassification.Evaluate(testPredictions);
|
||||||
|
Loading…
Reference in New Issue
Block a user