#lightgbm
Вопрос:
Я запустил пример/регрессию здесь https://github.com/microsoft/LightGBM/tree/master/examples/regression
Один простой вопрос заключается regression.train
regression.test
в том, в каком столбце находится функция и какой столбец используется в качестве целевого?
Ответ №1:
При использовании интерфейса командной строки LightGBM (CLI), если вы не укажете значение параметра label_column
in train.conf
, в качестве целевого будет использоваться первый столбец в обучающих данных.
Это можно увидеть в исходном коде DatasetLoader
класса LightGBM , используемого для создания Dataset
объекта LightGBM из файла.
label_idx_
(«столбец, используемый для целевого объекта») инициализируется значением 0: https://github.com/microsoft/LightGBM/blob/a8ee487aca35363fafa027e6b7695976045096b3/src/io/dataset_loader.cpp#L22label_idx_
переопределяется только в том случае, еслиlabel_column
указано в конфигурации: https://github.com/microsoft/LightGBM/blob/a8ee487aca35363fafa027e6b7695976045096b3/src/io/dataset_loader.cpp#L48