kaggle-metrics

Metrics for Kaggle competitions.

See wiki: Implemented metrics

Installation

python3.7 -m pip install git+https://github.com/krzjoa/kaggle-metrics.git

or:

python3.7 -m pip install kaggle_metrics

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)

Documentation

The full documentation is at http://kaggle-metrics.rtfd.org.