public ClusteringPredictionTransformer(IHostEnvironment env, TModel model, Schema inputSchema, string featureColumn, float threshold = 0f, string thresholdColumn = DefaultColumnNames.Score) : base(Contracts.CheckRef(env, nameof(env)).Register(nameof(ClusteringPredictionTransformer <TModel>)), model, inputSchema, featureColumn) { Host.CheckNonEmpty(thresholdColumn, nameof(thresholdColumn)); var schema = new RoleMappedSchema(inputSchema, null, featureColumn); var args = new ClusteringScorer.Arguments(); Scorer = new ClusteringScorer(Host, args, new EmptyDataView(Host, inputSchema), BindableMapper.Bind(Host, schema), schema); }
public ClusteringPredictionTransformer(IHostEnvironment env, ModelLoadContext ctx) : base(Contracts.CheckRef(env, nameof(env)).Register(nameof(ClusteringPredictionTransformer <TModel>)), ctx) { // *** Binary format *** // <base info> var schema = new RoleMappedSchema(TrainSchema, null, FeatureColumn); var args = new ClusteringScorer.Arguments(); Scorer = new ClusteringScorer(Host, args, new EmptyDataView(Host, TrainSchema), BindableMapper.Bind(Host, schema), schema); }
private ClusteringScorer(IHostEnvironment env, ClusteringScorer transform, IDataView newSource) : base(env, transform, newSource, RegistrationName) { }