juliadecisiontree模型-获取特征列表



我在Julia中使用通过算法创建的一些功能创建了一个决策树模型。

model_rest2 = DecisionTreeClassifier(n_subfeatures=0)
@time DecisionTree.fit!(model_rest2, convert(Matrix, df3[:,[16:45;49:50;52:81;83:end]]), df3[:,:type_1])

我似乎修改了功能构建步骤,因此在测试模型时,它现在没有运行,因为模型中的功能数量与测试集中可用的功能数量不同。有没有办法找到模型正在使用的功能列表,这样我就可以添加缺失的功能?

由于没有得到任何答案,我更改了操作数,并输入了所有列名,而不是数字列范围。这确保了我不会再陷入这个问题。

最新更新