
[ML] Catboost and DT - Sample Script from sklearn.tree import DecisionTreeRegressor from sklearn import tree dt = DecisionTreeRegressor(max_depth=3, min_samples_leaf=100) dtfeatures = ['is_female','is_seoul','age'] dt.fit(cust_prfl1[dtfeatures], cust_prfl1['amt']) # sklearn decision trees do not handle categorical data plt.figure(figsize=(16,8), dpi=80) tree.plot_tree(dt, feature_names=dtfeat..