pkiage commited on
Commit
8c3bb12
·
1 Parent(s): 4fb22e9

fix: reduce decision tree structure

Browse files

- smaller size enables render on heroku

Files changed (1) hide show
  1. src/models/xgboost_train_model.py +1 -1
src/models/xgboost_train_model.py CHANGED
@@ -43,7 +43,7 @@ def interpret_clf_xgbt_model(clf_xgbt_model):
43
  st.subheader("XGBoost Decision Tree Structure")
44
 
45
  (treexsize, treeysize,) = streamlit_chart_setting_height_width(
46
- "Chart Settings", 15, 10, "treexsize", "treeysize"
47
  )
48
 
49
  fig2 = plot_tree_gbt(treexsize, treeysize, clf_xgbt_model)
 
43
  st.subheader("XGBoost Decision Tree Structure")
44
 
45
  (treexsize, treeysize,) = streamlit_chart_setting_height_width(
46
+ "Chart Settings", 5, 5, "treexsize", "treeysize"
47
  )
48
 
49
  fig2 = plot_tree_gbt(treexsize, treeysize, clf_xgbt_model)