Skip to content
Snippets Groups Projects
Commit a843a1b3 authored by abp1g20's avatar abp1g20
Browse files

Update part_b.py

parent 438913fd
No related branches found
No related tags found
No related merge requests found
...@@ -29,9 +29,9 @@ test_data.columns = feature_names ...@@ -29,9 +29,9 @@ test_data.columns = feature_names
#X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=64) #X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=64)
# Instantiate the classifier # Instantiate the classifier
rf = ensemble.RandomForestClassifier() rf = ensemble.RandomForestClassifier(random_state=64)
gb = ensemble.HistGradientBoostingClassifier(max_iter=500, min_samples_leaf=17) gb = ensemble.HistGradientBoostingClassifier(max_iter=500, min_samples_leaf=17)
et = ensemble.ExtraTreesClassifier() et = ensemble.ExtraTreesClassifier(random_state=64)
model = ensemble.VotingClassifier( model = ensemble.VotingClassifier(
estimators=[('rf', rf), ('gb', gb), ('et', et)], estimators=[('rf', rf), ('gb', gb), ('et', et)],
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment