Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- train = pd.read_csv("data/train.csv")
- test = pd.read_csv("data/test.csv")
- X_y_train = xgb.DMatrix(data=train[['Pclass', 'Age', 'Fare', 'SibSp', 'Parch']],
- label=train['Survived'])
- X_test = xgb.DMatrix(data=test[['Pclass', 'Age', 'Fare', 'SibSp', 'Parch']])
- params = {
- 'base_score': np.mean(train['Survived']),
- 'eta': 0.1,
- 'max_depth': 3,
- 'gamma' :3,
- 'objective' :'reg:linear',
- 'eval_metric' :'mae'
- }
- model = xgb.train(params=params,
- dtrain=X_y_train,
- num_boost_round=3)
Add Comment
Please, Sign In to add comment