Generating Rules from Sklearn Decision Trees
Python
1| import pandas as pd 2| from sklearn.tree import DecisionTreeClassifier 3| from sklearn import tree 4| 5| #1. Create dummy dataframe for the example 6| df = pd.DataFrame({'X0':[0,2,2,2], 7| 'X1':[3,4,0,3], 8| 'target':[0,1,0,1]}) 9| 10| #2. Train model 11| model = DecisionTreeClassifier() 12| model.fit(df[['X0','X1']], df['target']) 13| 14| #3. Plot decision tree rules 15| tree.plot_tree(model) 16| plt.show()
130
118
115
108