This commit is contained in:
Guolin Ke 2017-08-07 21:12:28 +08:00
Родитель 09ac3485b5
Коммит 4e9b589bfd
4 изменённых файлов: 7004 добавлений и 5 удалений

Просмотреть файл

@ -1,4 +1,4 @@
data=../../examples/regression/regression.test
data=../data/categorical.data
input_model=LightGBM_model.txt

Просмотреть файл

@ -1,4 +1,6 @@
data=../../examples/regression/regression.train
data=../data/categorical.data
app=binary
num_trees=10

7000
tests/data/categorical.data Normal file

Разница между файлами не показана из-за своего большого размера Загрузить разницу

Просмотреть файл

@ -60,6 +60,3 @@ class TestBasic(unittest.TestCase):
for preds in zip(pred_early_stopping, pred_from_matr):
# scores likely to be different, but prediction should still be the same
self.assertEqual(preds[0] > 0, preds[1] > 0)
# check pmml
subprocess.call(['python', os.path.join(os.path.dirname(os.path.realpath(__file__)), '../../pmml/pmml.py'), 'model.txt'])