123 references to TrainUtils
Microsoft.ML.Data (74)
Commands\CrossValidationCommand.cs (13)
110TrainUtils.CheckTrainer(Host, args.Trainer, args.DataFile); 150if (ImplOptions.ContinueTrain && !TrainUtils.TryLoadPredictor(ch, Host, ImplOptions.InputModelFile, out inputPredictor)) 160string name = TrainUtils.MatchNameOrDefaultOrNull(ch, loader.Schema, nameof(ImplOptions.NameColumn), ImplOptions.NameColumn, DefaultColumnNames.Name); 271string label = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(ImplOptions.LabelColumn), ImplOptions.LabelColumn, DefaultColumnNames.Label); 272string features = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(ImplOptions.FeatureColumn), ImplOptions.FeatureColumn, DefaultColumnNames.Features); 273string weight = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(ImplOptions.WeightColumn), ImplOptions.WeightColumn, DefaultColumnNames.Weight); 274string name = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(ImplOptions.NameColumn), ImplOptions.NameColumn, DefaultColumnNames.Name); 275string group = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(ImplOptions.GroupColumn), ImplOptions.GroupColumn, DefaultColumnNames.GroupId); 277TrainUtils.AddNormalizerIfNeeded(env, ch, trainer, ref data, features, ImplOptions.NormalizeFeatures); 280var customCols = TrainUtils.CheckAndGenerateCustomColumns(ch, ImplOptions.CustomColumns); 297string group = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(ImplOptions.GroupColumn), ImplOptions.GroupColumn, DefaultColumnNames.GroupId); 523var predictor = TrainUtils.Train(host, ch, trainData, trainer, validData, 545TrainUtils.SaveModel(host, ch, file, predictor, rmd, _cmd);
Commands\EvaluateCommand.cs (9)
154string label = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(Arguments.LabelColumn), 156string group = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(Arguments.GroupColumn), 158string weight = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(Arguments.WeightColumn), 160var customCols = TrainUtils.CheckAndGenerateCustomColumns(ch, args.CustomColumns); 230string label = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(Arguments.LabelColumn), 232string group = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(Arguments.GroupColumn), 234string weight = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(Arguments.WeightColumn), 236string name = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(Arguments.NameColumn), 238var customCols = TrainUtils.CheckAndGenerateCustomColumns(ch, ImplOptions.CustomColumns);
Commands\ScoreCommand.cs (3)
116string feat = TrainUtils.MatchNameOrDefaultOrNull(ch, loader.Schema, 118string group = TrainUtils.MatchNameOrDefaultOrNull(ch, loader.Schema, 120var customCols = TrainUtils.CheckAndGenerateCustomColumns(ch, ImplOptions.CustomColumns);
Commands\TestCommand.cs (6)
101string label = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(ImplOptions.LabelColumn), 103string features = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(ImplOptions.FeatureColumn), 105string group = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(ImplOptions.GroupColumn), 107string weight = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(ImplOptions.WeightColumn), 109string name = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(ImplOptions.NameColumn), 111var customCols = TrainUtils.CheckAndGenerateCustomColumns(ch, ImplOptions.CustomColumns);
Commands\TrainCommand.cs (11)
101TrainUtils.CheckTrainer(Host, args.Trainer, args.DataFile); 145if (ImplOptions.ContinueTrain && !TrainUtils.TryLoadPredictor(ch, Host, ImplOptions.InputModelFile, out inputPredictor)) 152var label = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(Arguments.LabelColumn), _labelColumn, DefaultColumnNames.Label); 153var feature = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(Arguments.FeatureColumn), _featureColumn, DefaultColumnNames.Features); 154var group = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(Arguments.GroupColumn), _groupColumn, DefaultColumnNames.GroupId); 155var weight = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(Arguments.WeightColumn), _weightColumn, DefaultColumnNames.Weight); 156var name = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(Arguments.NameColumn), _nameColumn, DefaultColumnNames.Name); 158TrainUtils.AddNormalizerIfNeeded(Host, ch, trainer, ref view, feature, ImplOptions.NormalizeFeatures); 162var customCols = TrainUtils.CheckAndGenerateCustomColumns(ch, ImplOptions.CustomColumns); 200var predictor = TrainUtils.Train(Host, ch, data, trainer, validData, 204TrainUtils.SaveModel(Host, ch, file, predictor, data, cmd);
Commands\TrainTestCommand.cs (11)
90TrainUtils.CheckTrainer(Host, args.Trainer, args.DataFile); 128if (ImplOptions.ContinueTrain && !TrainUtils.TryLoadPredictor(ch, Host, ImplOptions.InputModelFile, out inputPredictor)) 135string label = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(Arguments.LabelColumn), 137string features = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(Arguments.FeatureColumn), 139string group = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(Arguments.GroupColumn), 141string weight = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(Arguments.WeightColumn), 143string name = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(Arguments.NameColumn), 146TrainUtils.AddNormalizerIfNeeded(Host, ch, trainer, ref trainPipe, features, ImplOptions.NormalizeFeatures); 149var customCols = TrainUtils.CheckAndGenerateCustomColumns(ch, ImplOptions.CustomColumns); 186var predictor = TrainUtils.Train(Host, ch, data, trainer, validData, 195TrainUtils.SaveModel(Host, ch, file, predictor, data, cmd);
EntryPoints\InputBase.cs (2)
83TrainUtils.AddNormalizerIfNeeded(host, ch, trainer, ref view, feature, input.NormalizeFeatures); 119var predictor = TrainUtils.Train(host, ch, cachedRoleMappedData, trainer, calibrator, maxCalibrationExamples);
EntryPoints\PredictorModelImpl.cs (1)
86TrainUtils.SaveModel(env, ch, stream, Predictor, roleMappedData);
EntryPoints\TransformModelImpl.cs (1)
179TrainUtils.SaveDataPipe(env, rep, _chain, blankLoader: true);
Evaluators\BinaryClassifierEvaluator.cs (3)
1535label = TrainUtils.MatchNameOrDefaultOrNull(host, schema, 1538weight = TrainUtils.MatchNameOrDefaultOrNull(host, schema, 1541name = TrainUtils.MatchNameOrDefaultOrNull(host, schema,
Evaluators\ClusteringEvaluator.cs (1)
867string features = TrainUtils.MatchNameOrDefaultOrNull(host, schema,
Evaluators\RankingEvaluator.cs (1)
1093string groupId = TrainUtils.MatchNameOrDefaultOrNull(host, schema,
Transforms\TrainAndScoreTransformer.cs (11)
93string feat = TrainUtils.MatchNameOrDefaultOrNull(env, input.Schema, 95string group = TrainUtils.MatchNameOrDefaultOrNull(env, input.Schema, 97var customCols = TrainUtils.CheckAndGenerateCustomColumns(env, args.CustomColumns); 238var customCols = TrainUtils.CheckAndGenerateCustomColumns(env, args.CustomColumns); 242var predictor = TrainUtils.Train(host, ch, data, trainer, null, 261feat = TrainUtils.MatchNameOrDefaultOrNull(ectx, schema, nameof(args.FeatureColumn), args.FeatureColumn, 263var label = TrainUtils.MatchNameOrDefaultOrNull(ectx, schema, nameof(args.LabelColumn), args.LabelColumn, 265group = TrainUtils.MatchNameOrDefaultOrNull(ectx, schema, nameof(args.GroupColumn), args.GroupColumn, 267var weight = TrainUtils.MatchNameOrDefaultOrNull(ectx, schema, nameof(args.WeightColumn), args.WeightColumn, 269var name = TrainUtils.MatchNameOrDefaultOrNull(ectx, schema, nameof(args.NameColumn), args.NameColumn, 271var customCols = TrainUtils.CheckAndGenerateCustomColumns(ectx, args.CustomColumns);
Utilities\ModelFileUtils.cs (1)
321return TrainUtils.CheckAndGenerateCustomColumns(env, list.ToArray());
Microsoft.ML.Ensemble (2)
EntryPoints\CreateEnsemble.cs (2)
332TrainUtils.SaveModel(env, ch, ms, null, data); 378TrainUtils.SaveModel(env, ch, ms, null, dataToCompare);
Microsoft.ML.EntryPoints (6)
ModelOperations.cs (3)
146var label = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(input.LabelColumnName), 149var feature = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(input.FeatureColumnName), 151var weight = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(input.ExampleWeightColumnName),
OneVersusAllMacro.cs (3)
122label = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(Arguments.LabelColumnName), 125var feature = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(Arguments.FeatureColumnName), 127var weight = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(Arguments.ExampleWeightColumnName),
Microsoft.ML.FastTree (1)
GamModelParameters.cs (1)
763TrainUtils.SaveModel(host, ch, file, _pred, _data);
Microsoft.ML.OnnxTransformerTest (2)
DnnImageFeaturizerTest.cs (1)
159TrainUtils.SaveModel(Env, Env.Start("saving"), ms, null, resultRoles);
OnnxTransformTests.cs (1)
202TrainUtils.SaveModel(Env, Env.Start("saving"), ms, null, resultRoles);
Microsoft.ML.TensorFlow.Tests (1)
TensorFlowEstimatorTests.cs (1)
130TrainUtils.SaveModel(Env, Env.Start("saving"), ms, null, resultRoles);
Microsoft.ML.Tests (29)
ImagesTests.cs (10)
195TrainUtils.SaveModel(env, ch, fh, null, new RoleMappedData(grey)); 356TrainUtils.SaveModel(env, ch, fh, null, new RoleMappedData(backToImages)); 431TrainUtils.SaveModel(env, ch, fh, null, new RoleMappedData(backToImages)); 507TrainUtils.SaveModel(env, ch, fh, null, new RoleMappedData(backToImages)); 582TrainUtils.SaveModel(env, ch, fh, null, new RoleMappedData(backToImages)); 657TrainUtils.SaveModel(env, ch, fh, null, new RoleMappedData(backToImages)); 733TrainUtils.SaveModel(env, ch, fh, null, new RoleMappedData(backToImages)); 808TrainUtils.SaveModel(env, ch, fh, null, new RoleMappedData(backToImages)); 884TrainUtils.SaveModel(env, ch, fh, null, new RoleMappedData(backToImages)); 958TrainUtils.SaveModel(env, ch, fh, null, new RoleMappedData(backToImages));
TermEstimatorTests.cs (1)
127TrainUtils.SaveModel(Env, Env.Start("saving"), ms, null, resultRoles);
Transformers\CategoricalHashTests.cs (1)
235TrainUtils.SaveModel(Env, ch, ms, null, resultRoles);
Transformers\CategoricalTests.cs (1)
328TrainUtils.SaveModel(Env, Env.Start("saving"), ms, null, resultRoles);
Transformers\CharTokenizeTests.cs (1)
67TrainUtils.SaveModel(Env, Env.Start("saving"), ms, null, resultRoles);
Transformers\ConvertTests.cs (1)
327TrainUtils.SaveModel(Env, Env.Start("saving"), ms, null, resultRoles);
Transformers\CopyColumnEstimatorTests.cs (1)
123TrainUtils.SaveModel(env, ((IHostEnvironment)env).Start("saving"), ms, null, resultRoles);
Transformers\FeatureSelectionTests.cs (2)
159TrainUtils.SaveModel(ML, Env.Start("saving"), ms, null, resultRoles); 219TrainUtils.SaveModel(ML, Env.Start("saving"), ms, null, resultRoles);
Transformers\HashTests.cs (1)
123TrainUtils.SaveModel(Env, Env.Start("saving"), ms, null, resultRoles);
Transformers\KeyToBinaryVectorEstimatorTest.cs (1)
160TrainUtils.SaveModel(Env, Env.Start("saving"), ms, null, resultRoles);
Transformers\KeyToVectorEstimatorTests.cs (1)
222TrainUtils.SaveModel(Env, Env.Start("saving"), ms, null, resultRoles);
Transformers\NAIndicatorTests.cs (1)
85TrainUtils.SaveModel(Env, Env.Start("saving"), ms, null, resultRoles);
Transformers\NAReplaceTests.cs (1)
180TrainUtils.SaveModel(Env, Env.Start("saving"), ms, null, resultRoles);
Transformers\NormalizerTests.cs (3)
743TrainUtils.SaveModel(ML, Env.Start("saving"), ms, null, resultRoles); 804TrainUtils.SaveModel(ML, Env.Start("saving"), ms, null, resultRoles); 864TrainUtils.SaveModel(ML, Env.Start("saving"), ms, null, resultRoles);
Transformers\RffTests.cs (1)
108TrainUtils.SaveModel(Env, Env.Start("saving"), ms, null, resultRoles);
Transformers\TextNormalizer.cs (1)
88TrainUtils.SaveModel(Env, Env.Start("saving"), ms, null, resultRoles);
Transformers\WordTokenizeTests.cs (1)
108TrainUtils.SaveModel(Env, Env.Start("saving"), ms, null, resultRoles);
Microsoft.ML.Transforms (8)
LearnerFeatureSelection.cs (8)
281var label = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(options.LabelColumn), options.LabelColumn, DefaultColumnNames.Label); 282var feature = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(options.FeatureColumn), options.FeatureColumn, DefaultColumnNames.Features); 283var group = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(options.GroupColumn), options.GroupColumn, DefaultColumnNames.GroupId); 284var weight = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(options.WeightColumn), options.WeightColumn, DefaultColumnNames.Weight); 285var name = TrainUtils.MatchNameOrDefaultOrNull(ch, schema, nameof(options.NameColumn), options.NameColumn, DefaultColumnNames.Name); 287TrainUtils.AddNormalizerIfNeeded(host, ch, trainer, ref view, feature, options.NormalizeFeatures); 291var customCols = TrainUtils.CheckAndGenerateCustomColumns(ch, options.CustomColumns); 294var predictor = TrainUtils.Train(host, ch, data, trainer, null,