Skip to content

krzjoa/kaggle-metrics

Folders and files

NameName
Last commit message
Last commit date

Latest commit

ย 

History

44 Commits
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 

Repository files navigation

kaggle-metrics

Python 3.7 PyPI version Documentation Status License: MIT

Metrics for Kaggle competitions.

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)