changed CatBoost model evaluate signature [#CLICKHOUSE-3305]

This commit is contained in:
Nikolai Kochetov 2017-10-26 17:10:51 +03:00
parent 4204a94b20
commit 092567af39

View File

@ -47,10 +47,10 @@ void FunctionModelEvaluate::executeImpl(Block & block, const ColumnNumbers & arg
auto model = models.getModel(name_col->getValue<String>()); auto model = models.getModel(name_col->getValue<String>());
Columns columns; ConstColumnPlainPtrs columns;
columns.reserve(arguments.size()); columns.reserve(arguments.size());
for (auto i : ext::range(1, arguments.size())) for (auto i : ext::range(1, arguments.size()))
columns.push_back(block.getByPosition(arguments[i]).column); columns.push_back(block.getByPosition(arguments[i]).column.get());
block.getByPosition(result).column = model->evaluate(columns); block.getByPosition(result).column = model->evaluate(columns);
} }