2 writes to TrainSchema
Microsoft.ML.Data (2)
Scorers\PredictionTransformer.cs (2)
91TrainSchema = trainSchema; 131TrainSchema = loader.Schema;
17 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);