保存决策树模型以供以后在 Julia 中应用

Question

我已经使用 DecisionTree 模块在 Julia 中训练了一个经过修剪的决策树模型。我现在想保存此模型以供以后在其他数据集上使用。

我试过使用 writetable() 将模型转换为数据数组以供导出，我试过使用 writedlm() 导出，但这些都不起作用。当我查看模型的类型时，我发现它是 DecisionTree.Node 类型。我不知道如何使用它，也无法将其发送到 export/save。

In:DataFrame(PrunedModel)
Out:LoadError: MethodError: `convert` has no method matching convert(::Type{DataFrames.DataFrame}, ::DecisionTree.Node)
This may have arisen from a call to the constructor DataFrames.DataFrame(...),
since type constructors fall back to convert methods.
Closest candidates are:
  call{T}(::Type{T}, ::Any)
  convert(::Type{DataFrames.DataFrame}, !Matched::Array{T,2})
  convert(::Type{DataFrames.DataFrame}, !Matched::Dict{K,V})
  ...
while loading In[22], in expression starting on line 1 
 in call at essentials.jl:56

In:typeof(PrunedModel)
Out:DecisionTree.Node

关于如何保存此模型供以后使用的任何想法？

Answer 1

如果我正确理解这是一个 Julia 对象，您应该尝试使用 JLD.jl 包将对象保存到磁盘并重新加载，同时保留类型信息。

保存决策树模型以供以后在 Julia 中应用

Saving a decision tree model for later application in Julia

export

decision-tree

julia