public override void Save(string path) { if (TrainedModel == null) { throw new InvalidOperationException("Must train/load a model before saving"); } #if ML_LEGACY TrainedModel.WriteAsync(path).Wait(); #else // save using (var stream = File.Create(path)) { Context.Model.Save(TrainedModel, InputSchema, stream); } #endif }
public override void Save(string path) { if (TrainedModel == null) { throw new Exception("Must initialize the model before calling"); } lock (TrainedModel) { #if ML_LEGACY TrainedModel.WriteAsync(path).Wait(); #else // save using (var stream = File.Create(path)) { TrainedModel.SaveTo(Context, stream); } #endif } }