Exemplo n.º 1
0
        public XGBoostScalarRowMapperBase(RoleMappedSchema schema, XGBoostPredictorBase <TOutput> parent, IHostEnvironment env, ISchema outputSchema)
        {
            Contracts.AssertValue(env, "env");
            env.AssertValue(schema, "schema");
            env.AssertValue(parent, "parent");
            env.AssertValue(schema.Feature, "schema");

            // REVIEW xadupre: only one feature columns is allowed.
            // This should be revisited in the future.
            // XGBoost has plans for others types.
            // Look at https://github.com/dmlc/xgboost/issues/874.
            env.Check(schema.Feature != null, "Unexpected number of feature columns, 1 expected.");

            _parent = parent;
            var columns = new[] { schema.Feature };
            var fc      = new[] { new KeyValuePair <RoleMappedSchema.ColumnRole, string>(RoleMappedSchema.ColumnRole.Feature, columns[0].Name) };

            _inputSchema  = new RoleMappedSchema(schema.Schema, fc);
            _outputSchema = outputSchema;

            _inputCols = new List <int>();
            foreach (var kvp in columns)
            {
                int index;
                if (schema.Schema.TryGetColumnIndex(kvp.Name, out index))
                {
                    _inputCols.Add(index);
                }
                else
                {
                    Contracts.Assert(false);
                }
            }

            _booster = _parent.GetBooster();
        }
 public XGBoostScalarRowMapper(RoleMappedSchema schema, XGBoostPredictorBase <Float> parent, IHostEnvironment env, ISchema outputSchema)
     : base(schema, parent, env, outputSchema)
 {
     env.CheckParam(outputSchema.ColumnCount == 1, nameof(outputSchema));
     env.CheckParam(outputSchema.GetColumnType(0).IsNumber, nameof(outputSchema));
 }