Advertisement
Guest User

Untitled

a guest
Oct 18th, 2019
80
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.54 KB | None | 0 0
  1. # initialise the base models
  2. model_1 = tree.DecisionTreeClassifier(random_state=1)
  3. model_2 = KNeighborsClassifier()
  4.  
  5. # train base models and create new featurs
  6. test_pred_1 ,train_pred_1=Stacking(model=model_1,n_fold=10, train=x_train,test=x_test,y=y_train)
  7. test_pred_2 ,train_pred_2=Stacking(model=model_2,n_fold=10,train=x_train,test=x_test,y=y_train)
  8.  
  9. # convert into dataframe for later use
  10. train_pred_1=pd.DataFrame(train_pred_1)
  11. test_pred_1=pd.DataFrame(test_pred_1)
  12. train_pred_2=pd.DataFrame(train_pred_2)
  13. test_pred_2=pd.DataFrame(test_pred_2)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement