Skip to content

Instantly share code, notes, and snippets.

@Abhayparashar31
Created September 15, 2022 16:01
Show Gist options
  • Save Abhayparashar31/8ec67126131e695fafa86ff93a5b0c1b to your computer and use it in GitHub Desktop.
Save Abhayparashar31/8ec67126131e695fafa86ff93a5b0c1b to your computer and use it in GitHub Desktop.
from sklearn.neighbors import KNeighborsClassifier
from sklearn.tree import DecisionTreeClassifier
from sklearn.svm import SVC
from sklearn.linear_model import LogisticRegression
from sklearn.ensemble import StackingClassifier
base_learners = [
('l1', KNeighborsClassifier()),
('l2', DecisionTreeClassifier()),
('l3',SVC(gamma=2, C=1)))
]
model = StackingClassifier(estimators=base_learners, final_estimator=LogisticRegression(),cv=5)
model.fit(X_train, y_train)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment