From cf186c2ef15ce5555c32d9117ddf7ac50c36b9ef Mon Sep 17 00:00:00 2001 From: Kosterix08 Date: Tue, 18 Jun 2024 23:24:03 +0200 Subject: [PATCH] add graphviz to predict client --- decisiontree.py | 20 +++++++++++--------- 1 file changed, 11 insertions(+), 9 deletions(-) diff --git a/decisiontree.py b/decisiontree.py index 8e5513f..5532e7a 100644 --- a/decisiontree.py +++ b/decisiontree.py @@ -34,7 +34,7 @@ X_test = encoder.transform(X_test) #print(X_train.head()) #print(X_test.head()) -clf_en = DecisionTreeClassifier(criterion='entropy', max_depth=3, random_state=0) +clf_en = DecisionTreeClassifier(criterion='entropy', max_depth=200, random_state=0) clf_en.fit(X_train, y_train) y_pred_en = clf_en.predict(X_test) @@ -46,12 +46,6 @@ y_pred_train_en = clf_en.predict(X_train) #print('Training set score: {:.4f}'.format(clf_en.score(X_train, y_train))) #print('Test set score: {:.4f}'.format(clf_en.score(X_test, y_test))) -dot_data = tree.export_graphviz(clf_en, out_file=None, - feature_names=X_train.columns, - class_names=y_train.unique(), - filled=True, rounded=True, - special_characters=True) - #nowy klient testowo """new_client = { "Wrinkles": random.choice(['Yes', 'No']), @@ -83,6 +77,15 @@ def predict_client(client_data,glasses): print('Training set score: {:.4f}'.format(clf_en.score(X_train, y_train))) print('Test set score: {:.4f}'.format(clf_en.score(X_test, y_test))) + + dot_data = tree.export_graphviz(clf_en, out_file=None, + feature_names=X_train.columns, + class_names=y_train.unique(), + filled=True, rounded=True, + special_characters=True) + + graph = graphviz.Source(dot_data) + graph.render("decision_tree", format='png') return prediction[0] @@ -90,5 +93,4 @@ def predict_client(client_data,glasses): #print(new_client_df) #print("Prediction:", prediction[0]) -#graph = graphviz.Source(dot_data) -#graph.render("decision_tree", format='png') +