Guest User

Untitled

a guest
Apr 23rd, 2018
87
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.89 KB | None | 0 0
  1. from sklearn.preprocessing import StandardScaler, MinMaxScaler
  2. from sklearn.pipeline import Pipeline
  3. from sklearn.datasets import make_regression
  4. from sklearn.tree import DecisionTreeRegressor
  5. from sklearn.metrics import mean_squared_error, r2_score
  6. from operator import itemgetter
  7.  
  8. import graphviz
  9. from sklearn.tree import export_graphviz
  10. from os import system
  11.  
  12.  
  13.  
  14.  
  15. X, y = make_regression (n_samples = 1000, n_features =5)
  16.  
  17. best_tree = DecisionTreeRegressor(criterion='mse', max_depth=20).fit (X,y)
  18. export_graphviz(decision_tree=best_tree.tree_, out_file='noP.dot')
  19. os.system("dot -Tpng noP.dot -o "+ 'noP.png')
  20.  
  21. > for i in range(best_tree.tree_.node_count):
  22. > n_value = best_tree.tree_.
  23. > if (n_value < 10):
  24. > #print('do i get here', n_value, best_tree.tree_.children_left[i])
  25. > best_tree.tree_.children_left[i]=-1
  26. > best_tree.tree_.children_right[i]=-1
Add Comment
Please, Sign In to add comment