# view
查看回归树
函数库: TyMachineLearning
# 语法
view(mdf)
# 说明
view(mdf)返回拟合的回归树模型mdf的决策树文本描述。
# 示例
通过view可视化回归树模型
加载数据集，tree_x中的三个变量特征分别记录着排放量、马力以及重量的信息，tree_y记录着每加仑的英里数MPG。
using TyMachineLearning
using CSV
using DataFrames
file1 = joinpath(pkgdir(TyMachineLearning), "data/Regression/tree_x.csv")
file2 = joinpath(pkgdir(TyMachineLearning), "data/Regression/tree_y.csv")
fgdata_x = CSV.read(file1, DataFrame; header=0)
fgdata_y = CSV.read(file2, DataFrame; header=0)
fgdata_x1 = Matrix(fgdata_x)
fgdata_y1 = Array(fgdata_y)
X = DataFrame([fgdata_x1 fgdata_y1], :auto)
rename!(X, [:x4 => :y])
X_pre, y_pre = dropna(X)
显示样本数据X的前5行数据。
X[1:5,:]
5×4 DataFrame
Row │ x1 x2 x3 y
│ Float64 Float64 Float64 Float64
─────┼────────────────────────────────────
1 │ 307.0 130.0 3504.0 18.0
2 │ 350.0 165.0 3693.0 15.0
3 │ 318.0 150.0 3436.0 18.0
4 │ 304.0 150.0 3433.0 16.0
5 │ 302.0 140.0 3449.0 17.0
对该数据集进行回归树fitrtree模型拟合。
mdf = fitrtree(X_pre, y_pre;max_depth=2)
PyObject DecisionTreeRegressor(max_depth=2, min_weight_fraction_leaf=0, random_state=1)
可视化拟合的回归树。
TyMachineLearning.view(mdf)
# 输入参数
mdf - 回归树模型
模型对象
回归拟合二叉决策树模型mdf。
数据类型: 模型对象