kaggle-metrics

Python version PyPI version Documentation Status Licence

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:

Module Index
Search Page