kaggle-metrics¶
Metrics for Kaggle competitions.
Installation¶
You can install this module directly from GitHub repo with command:
or as a PyPI package
Usage¶
from xgboost import XGBRegressor
import kaggle_metrics as km
X_train, y_train, X_test, y_test = get_data()
# Train
clf = XGBRegressor()
clf.fit(X_train, y_train)
# Get predictions
y_pred = clf.predict(X_test)
# Evaluate with kaggle-metrics
km.rmse(y_test, y_pred)
Contents: