Guest User

Untitled

a guest
Jul 18th, 2018
96
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.59 KB | None | 0 0
  1. train = pd.read_csv("data/train.csv")
  2. test = pd.read_csv("data/test.csv")
  3. X_y_train = xgb.DMatrix(data=train[['Pclass', 'Age', 'Fare', 'SibSp', 'Parch']],
  4. label=train['Survived'])
  5. X_test = xgb.DMatrix(data=test[['Pclass', 'Age', 'Fare', 'SibSp', 'Parch']])
  6.  
  7. params = {
  8. 'base_score': np.mean(train['Survived']),
  9. 'eta': 0.1,
  10. 'max_depth': 3,
  11. 'gamma' :3,
  12. 'objective' :'reg:linear',
  13. 'eval_metric' :'mae'
  14. }
  15. model = xgb.train(params=params,
  16. dtrain=X_y_train,
  17. num_boost_round=3)
Add Comment
Please, Sign In to add comment