Skip to content

Commit

Permalink
Let's model - added decision tree plot
Browse files Browse the repository at this point in the history
  • Loading branch information
Dor Meir authored and Dor Meir committed Jul 30, 2023
1 parent c0c53b8 commit 396d061
Show file tree
Hide file tree
Showing 2 changed files with 45 additions and 15 deletions.
56 changes: 42 additions & 14 deletions Pipeline.ipynb

Large diffs are not rendered by default.

4 changes: 3 additions & 1 deletion model_metrics.csv
Original file line number Diff line number Diff line change
Expand Up @@ -10,4 +10,6 @@ LogisticRegression()_engineered_features,2023-07-30 08:13:04,LogisticRegression(
DecisionTreeClassifier(),2023-07-30 08:50:18,DecisionTreeClassifier(),"{'ccp_alpha': 0.0, 'class_weight': None, 'criterion': 'gini', 'max_depth': None, 'max_features': None, 'max_leaf_nodes': None, 'min_impurity_decrease': 0.0, 'min_samples_leaf': 1, 'min_samples_split': 2, 'min_weight_fraction_leaf': 0.0, 'random_state': None, 'splitter': 'best'}",204,836,122,81,755,82,0.8432692307692308,0.1961538461538461,0.6009852216748769,3.063846228146432,0.5980392156862745,0.5995085995085996,0.7505746317665823,0.902031063321386,1.122143906524212,0.90311004784689,0.9025702331141664,"{'fixed_acidity': 0.059, 'volatile_acidity': 0.143, 'citric_acid': 0.054, 'residual_sugar': 0.092, 'chlorides': 0.058, 'free_sulfur_dioxide': 0.05, 'total_sulfur_dioxide': 0.078, 'density': 0.071, 'pH': 0.064, 'sulphates': 0.071, 'alcohol': 0.262, 'type_white': 0.0}"
DecisionTreeClassifier()_engineered_features,2023-07-30 08:50:18,DecisionTreeClassifier(),"{'ccp_alpha': 0.0, 'class_weight': None, 'criterion': 'gini', 'max_depth': None, 'max_features': None, 'max_leaf_nodes': None, 'min_impurity_decrease': 0.0, 'min_samples_leaf': 1, 'min_samples_split': 2, 'min_weight_fraction_leaf': 0.0, 'random_state': None, 'splitter': 'best'}",204,836,100,80,756,104,0.823076923076923,0.1961538461538461,0.5555555555555556,2.832244008714597,0.4901960784313725,0.5208333333333333,0.6972511492635332,0.8790697674418605,1.0935796149994437,0.9043062200956936,0.891509433962264,"{'fixed_acidity': 0.05, 'volatile_acidity': 0.086, 'citric_acid': 0.04, 'residual_sugar': 0.067, 'chlorides': 0.047, 'free_sulfur_dioxide': 0.03, 'total_sulfur_dioxide': 0.047, 'density': 0.073, 'pH': 0.051, 'sulphates': 0.072, 'alcohol': 0.265, 'type_white': 0.0, 'fixed_acidity_is_outlier': 0.0, 'volatile_acidity_is_outlier': 0.0, 'citric_acid_is_outlier': 0.0, 'residual_sugar_is_outlier': 0.0, 'chlorides_is_outlier': 0.001, 'free_sulfur_dioxide_is_outlier': 0.0, 'total_sulfur_dioxide_is_outlier': 0.0, 'density_is_outlier': 0.0, 'pH_is_outlier': 0.001, 'sulphates_is_outlier': 0.001, 'alcohol_is_outlier': 0.0, 'chlorides_X_volatile_acidity': 0.073, 'total_acidity/free_sulfur_dioxide': 0.039, 'free_sulfur_dioxide/total_sulfur_dioxide': 0.059}"
RandomForestClassifier(),2023-07-30 09:21:57,RandomForestClassifier(),"{'bootstrap': True, 'ccp_alpha': 0.0, 'class_weight': None, 'criterion': 'gini', 'max_depth': None, 'max_features': 'sqrt', 'max_leaf_nodes': None, 'max_samples': None, 'min_impurity_decrease': 0.0, 'min_samples_leaf': 1, 'min_samples_split': 2, 'min_weight_fraction_leaf': 0.0, 'n_estimators': 100, 'n_jobs': None, 'oob_score': False, 'random_state': None, 'verbose': 0, 'warm_start': False}",204,836,119,49,787,85,0.8711538461538462,0.1961538461538461,0.7083333333333334,3.611111111111111,0.5833333333333334,0.6397849462365591,0.9006619992494604,0.9025229357798163,1.122755805276327,0.9413875598086124,0.9215456674473068,"{'fixed_acidity': 0.059, 'volatile_acidity': 0.098, 'citric_acid': 0.086, 'residual_sugar': 0.075, 'chlorides': 0.083, 'free_sulfur_dioxide': 0.065, 'total_sulfur_dioxide': 0.076, 'density': 0.107, 'pH': 0.071, 'sulphates': 0.077, 'alcohol': 0.199, 'type_white': 0.002}"
RandomForestClassifier()_engineered_features,2023-07-30 09:21:59,RandomForestClassifier(),"{'bootstrap': True, 'ccp_alpha': 0.0, 'class_weight': None, 'criterion': 'gini', 'max_depth': None, 'max_features': 'sqrt', 'max_leaf_nodes': None, 'max_samples': None, 'min_impurity_decrease': 0.0, 'min_samples_leaf': 1, 'min_samples_split': 2, 'min_weight_fraction_leaf': 0.0, 'n_estimators': 100, 'n_jobs': None, 'oob_score': False, 'random_state': None, 'verbose': 0, 'warm_start': False}",204,836,114,41,795,90,0.8740384615384615,0.19615384615384615,0.7354838709677419,3.749525616698292,0.5588235294117647,0.6350974930362117,0.9064610892203772,0.8983050847457628,1.117508717865542,0.9509569377990431,0.9238814642649622,"{'fixed_acidity': 0.047, 'volatile_acidity': 0.066, 'citric_acid': 0.068, 'residual_sugar': 0.063, 'chlorides': 0.067, 'free_sulfur_dioxide': 0.044, 'total_sulfur_dioxide': 0.059, 'density': 0.089, 'pH': 0.056, 'sulphates': 0.066, 'alcohol': 0.17, 'type_white': 0.003, 'fixed_acidity_is_outlier': 0.001, 'volatile_acidity_is_outlier': 0.0, 'citric_acid_is_outlier': 0.001, 'residual_sugar_is_outlier': 0.0, 'chlorides_is_outlier': 0.001, 'free_sulfur_dioxide_is_outlier': 0.0, 'total_sulfur_dioxide_is_outlier': 0.0, 'density_is_outlier': 0.0, 'pH_is_outlier': 0.001, 'sulphates_is_outlier': 0.0, 'alcohol_is_outlier': 0.0, 'chlorides_X_volatile_acidity': 0.081, 'total_acidity/free_sulfur_dioxide': 0.056, 'free_sulfur_dioxide/total_sulfur_dioxide': 0.061}"
RandomForestClassifier()_engineered_features,2023-07-30 09:21:59,RandomForestClassifier(),"{'bootstrap': True, 'ccp_alpha': 0.0, 'class_weight': None, 'criterion': 'gini', 'max_depth': None, 'max_features': 'sqrt', 'max_leaf_nodes': None, 'max_samples': None, 'min_impurity_decrease': 0.0, 'min_samples_leaf': 1, 'min_samples_split': 2, 'min_weight_fraction_leaf': 0.0, 'n_estimators': 100, 'n_jobs': None, 'oob_score': False, 'random_state': None, 'verbose': 0, 'warm_start': False}",204,836,114,41,795,90,0.8740384615384615,0.1961538461538461,0.7354838709677419,3.749525616698292,0.5588235294117647,0.6350974930362117,0.9064610892203772,0.8983050847457628,1.117508717865542,0.9509569377990432,0.9238814642649622,"{'fixed_acidity': 0.047, 'volatile_acidity': 0.066, 'citric_acid': 0.068, 'residual_sugar': 0.063, 'chlorides': 0.067, 'free_sulfur_dioxide': 0.044, 'total_sulfur_dioxide': 0.059, 'density': 0.089, 'pH': 0.056, 'sulphates': 0.066, 'alcohol': 0.17, 'type_white': 0.003, 'fixed_acidity_is_outlier': 0.001, 'volatile_acidity_is_outlier': 0.0, 'citric_acid_is_outlier': 0.001, 'residual_sugar_is_outlier': 0.0, 'chlorides_is_outlier': 0.001, 'free_sulfur_dioxide_is_outlier': 0.0, 'total_sulfur_dioxide_is_outlier': 0.0, 'density_is_outlier': 0.0, 'pH_is_outlier': 0.001, 'sulphates_is_outlier': 0.0, 'alcohol_is_outlier': 0.0, 'chlorides_X_volatile_acidity': 0.081, 'total_acidity/free_sulfur_dioxide': 0.056, 'free_sulfur_dioxide/total_sulfur_dioxide': 0.061}"
DecisionTreeClassifier(),2023-07-30 10:00:06,DecisionTreeClassifier(),"{'ccp_alpha': 0.0, 'class_weight': None, 'criterion': 'gini', 'max_depth': None, 'max_features': None, 'max_leaf_nodes': None, 'min_impurity_decrease': 0.0, 'min_samples_leaf': 1, 'min_samples_split': 2, 'min_weight_fraction_leaf': 0.0, 'random_state': None, 'splitter': 'best'}",204,836,126,78,758,78,0.85,0.1961538461538461,0.6176470588235294,3.1487889273356404,0.6176470588235294,0.6176470588235294,0.7621728117084154,0.9066985645933014,1.127950367436643,0.9066985645933014,0.9066985645933014,"{'fixed_acidity': 0.057, 'volatile_acidity': 0.139, 'citric_acid': 0.047, 'residual_sugar': 0.094, 'chlorides': 0.063, 'free_sulfur_dioxide': 0.057, 'total_sulfur_dioxide': 0.067, 'density': 0.071, 'pH': 0.068, 'sulphates': 0.068, 'alcohol': 0.267, 'type_white': 0.002}"
DecisionTreeClassifier()_engineered_features,2023-07-30 10:00:06,DecisionTreeClassifier(),"{'ccp_alpha': 0.0, 'class_weight': None, 'criterion': 'gini', 'max_depth': None, 'max_features': None, 'max_leaf_nodes': None, 'min_impurity_decrease': 0.0, 'min_samples_leaf': 1, 'min_samples_split': 2, 'min_weight_fraction_leaf': 0.0, 'random_state': None, 'splitter': 'best'}",204,836,100,90,746,104,0.8134615384615385,0.19615384615384615,0.5263157894736842,2.683178534571723,0.49019607843137253,0.5076142131979695,0.6912702880195141,0.8776470588235294,1.0918097382493668,0.8923444976076556,0.8849347568208779,"{'fixed_acidity': 0.044, 'volatile_acidity': 0.084, 'citric_acid': 0.039, 'residual_sugar': 0.069, 'chlorides': 0.042, 'free_sulfur_dioxide': 0.034, 'total_sulfur_dioxide': 0.051, 'density': 0.073, 'pH': 0.052, 'sulphates': 0.071, 'alcohol': 0.265, 'type_white': 0.001, 'fixed_acidity_is_outlier': 0.0, 'volatile_acidity_is_outlier': 0.0, 'citric_acid_is_outlier': 0.0, 'residual_sugar_is_outlier': 0.0, 'chlorides_is_outlier': 0.001, 'free_sulfur_dioxide_is_outlier': 0.0, 'total_sulfur_dioxide_is_outlier': 0.0, 'density_is_outlier': 0.0, 'pH_is_outlier': 0.002, 'sulphates_is_outlier': 0.0, 'alcohol_is_outlier': 0.0, 'chlorides_X_volatile_acidity': 0.073, 'total_acidity/free_sulfur_dioxide': 0.038, 'free_sulfur_dioxide/total_sulfur_dioxide': 0.062}"

0 comments on commit 396d061

Please sign in to comment.