Skip to content

Use Quadratic Weighted Kappa

from sklearn.datasets import make_multilabel_classification
from sklearn.model_selection import train_test_split
from sklearn.metrics import cohen_kappa_score
from bokbokbok.eval_metrics.classification import QuadraticWeightedKappaMetric
from bokbokbok.utils import clip_sigmoid

X, y = make_multilabel_classification(n_samples=1000, 
                                      n_features=10,
                                      n_classes=2,
                                      n_labels=1,
                                      random_state=41114)
y = y.sum(axis=1)
X_train, X_valid, y_train, y_valid = train_test_split(X, 
                                                      y, 
                                                      test_size=0.25, 
                                                      random_state=41114)

Usage in LightGBM

import lightgbm as lgb

train = lgb.Dataset(X_train, y_train)
valid = lgb.Dataset(X_valid, y_valid, reference=train)
params = {
     'n_estimators': 300,
     'seed': 41114,
     'n_jobs': 8,
     'learning_rate': 0.1,
    'objective':'multiclass',
    'num_class': 3
   }

clf = lgb.train(params=params,
                train_set=train,
                valid_sets=[train, valid],
                valid_names=['train','valid'],
                feval=QuadraticWeightedKappaMetric(),
                early_stopping_rounds=100)

Usage in XGBoost

import xgboost as xgb

dtrain = xgb.DMatrix(X_train, y_train)
dvalid = xgb.DMatrix(X_valid, y_valid)

params = {
     'seed': 41114,
     'learning_rate': 0.1,
    'disable_default_eval_metric': 1,
    'objective': 'multi:softprob',
    'num_class': 3
   }

bst = xgb.train(params,
          dtrain=dtrain,
          num_boost_round=300,
          early_stopping_rounds=100,
          verbose_eval=10,
          feval=QuadraticWeightedKappaMetric(XGBoost=True),
          maximize=True,
          evals=[(dtrain, 'dtrain'), (dvalid, 'dvalid')])