2 writes to TrainSchema
Microsoft.ML.Data (2)
Scorers\PredictionTransformer.cs (2)
91TrainSchema = trainSchema; 131TrainSchema = loader.Schema;
33 references to TrainSchema
Microsoft.ML.Data (17)
Scorers\PredictionTransformer.cs (15)
180DataSaverUtils.SaveDataView(ch, saver, new EmptyDataView(Host, TrainSchema), writer.BaseStream); 248else if (!TrainSchema.TryGetColumnIndex(FeatureColumnName, out int col)) 251FeatureColumnType = TrainSchema[col].Type; 263else if (!TrainSchema.TryGetColumnIndex(FeatureColumnName, out int col)) 266FeatureColumnType = TrainSchema[col].Type; 306var schema = new RoleMappedSchema(TrainSchema, null, FeatureColumnName); 307return new GenericScorer(Host, new GenericScorer.Arguments(), new EmptyDataView(Host, TrainSchema), BindableMapper.Bind(Host, schema), schema); 348var schema = new RoleMappedSchema(TrainSchema, null, FeatureColumnName); 350Scorer = new BinaryClassifierScorer(Host, args, new EmptyDataView(Host, TrainSchema), BindableMapper.Bind(Host, schema), schema); 440var schema = new RoleMappedSchema(TrainSchema, LabelColumnName, FeatureColumnName); 442Scorer = new BinaryClassifierScorer(Host, args, new EmptyDataView(Host, TrainSchema), BindableMapper.Bind(Host, schema), schema); 532var schema = new RoleMappedSchema(TrainSchema, _trainLabelColumn, FeatureColumnName); 534Scorer = new MulticlassClassificationScorer(Host, args, new EmptyDataView(Host, TrainSchema), BindableMapper.Bind(Host, schema), schema); 686var schema = new RoleMappedSchema(TrainSchema, null, FeatureColumnName); 688Scorer = new ClusteringScorer(Host, args, new EmptyDataView(Host, TrainSchema), BindableMapper.Bind(Host, schema), schema);
TrainCatalog.cs (2)
296return new BinaryPredictionTransformer<TModel>(Environment, model.Model, model.TrainSchema, model.FeatureColumnName, threshold, model.ThresholdColumn); 790return new AnomalyPredictionTransformer<TModel>(Environment, model.Model, model.TrainSchema, model.FeatureColumnName, threshold, model.ThresholdColumn);
Microsoft.ML.FastTree (4)
TreeEnsembleFeaturizationTransformer.cs (4)
116_featureDetachedColumn = new DataViewSchema.DetachedColumn(TrainSchema[featureColumnName]); 134var roleMappedSchema = MakeFeatureRoleMappedSchema(TrainSchema); 135Scorer = new GenericScorer(Host, _scorerArgs, new EmptyDataView(Host, TrainSchema), BindableMapper.Bind(Host, roleMappedSchema), roleMappedSchema); 168DataSaverUtils.SaveDataView(ch, saver, new EmptyDataView(Host, TrainSchema), writer.BaseStream);
Microsoft.ML.Recommender (7)
MatrixFactorizationPredictor.cs (7)
474var schema = new RoleMappedSchema(TrainSchema, roles); 493if (!TrainSchema.TryGetColumnIndex(MatrixColumnIndexColumnName, out int xCol)) 495MatrixColumnIndexColumnType = TrainSchema[xCol].Type; 497if (!TrainSchema.TryGetColumnIndex(MatrixRowIndexColumnName, out int yCol)) 499MatrixRowIndexColumnType = TrainSchema[yCol].Type; 505Scorer = new GenericScorer(Host, args, new EmptyDataView(Host, TrainSchema), BindableMapper.Bind(Host, schema), schema); 541DataSaverUtils.SaveDataView(ch, saver, new EmptyDataView(Host, TrainSchema), writer.BaseStream);
Microsoft.ML.StandardTrainers (5)
FactorizationMachine\FieldAwareFactorizationMachineModelParameters.cs (5)
350if (!TrainSchema.TryGetColumnIndex(featureColumns[i], out int col)) 352featureColumnTypes[i] = TrainSchema[col].Type; 364Scorer = new BinaryClassifierScorer(Host, args, new EmptyDataView(Host, TrainSchema), BindableMapper.Bind(Host, schema), schema); 410DataSaverUtils.SaveDataView(ch, saver, new EmptyDataView(Host, TrainSchema), writer.BaseStream); 427var schema = new RoleMappedSchema(TrainSchema, roles);