Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

DOC/TST: Fix Python sklearn dep

Browse Source
This commit is contained in:
sinhrks
2016-05-01 17:10:11 +09:00
parent 2f2ad21de4
commit 9da2f3e613
10 changed files with 130 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/python/test_eval_metrics.py
View File

@@ -1,8 +1,6 @@
import xgboost as xgb
import xgboost.testing as tm
import numpy as np
from sklearn.cross_validation import train_test_split
from sklearn.metrics import mean_squared_error
from sklearn.datasets import load_digits
import unittest
rng = np.random.RandomState(1337)
@@ -42,16 +40,26 @@ class TestEvalMetrics(unittest.TestCase):
return [('error', float(sum(labels != (preds > 0.0))) / len(labels))]
def evalerror_03(self, preds, dtrain):
tm._skip_if_no_sklearn()
from sklearn.metrics import mean_squared_error
labels = dtrain.get_label()
return [('rmse', mean_squared_error(labels, preds)),
('error', float(sum(labels != (preds > 0.0))) / len(labels))]
def evalerror_04(self, preds, dtrain):
tm._skip_if_no_sklearn()
from sklearn.metrics import mean_squared_error
labels = dtrain.get_label()
return [('error', float(sum(labels != (preds > 0.0))) / len(labels)),
('rmse', mean_squared_error(labels, preds))]
def test_eval_metrics(self):
tm._skip_if_no_sklearn()
from sklearn.cross_validation import train_test_split
from sklearn.datasets import load_digits
digits = load_digits(2)
X = digits['data']
y = digits['target']