Похоже, что метод trees_to_dataframe
на объекте lightgbm.Booster
отсутствует. https://lightgbm.readthedocs.io/en/latest/pythonapi/lightgbm.Booster.html#lightgbm .Booster.trees_to_dataframe .
Следующий код вызывает AttributeError
в последней строке.
import numpy as np
import pandas as pd
import lightgbm as lgb
number_of_classes = 5
x = np.random.random((100, 100))
y = np.random.randint(low=0, high=number_of_classes, size=100)
data = lgb.Dataset(x, y)
parameters = {
'objective' : 'multiclass',
'num_class' : number_of_classes,
'metric' : ['multi_logloss', ],
}
bst = lgb.train(parameters, data)
assert type(bst) == lgb.Booster
bst.trees_to_dataframe()