search

Xtra-Computing / thundergbm

ThunderGBM: Fast GBDTs and Random Forests on GPUs
Apache License 2.0
689 stars 85 forks source link

Jupyter Notebook Kernel died after TGBMRegressor().fit(X_train, y_train) #29

Closed lyghter closed 4 years ago

lyghter commented 4 years ago

import json import numpy as np import pandas as pd import thundergbm

INPUT_DIR = 'https://raw.githubusercontent.com/ArtyomSalnikov/tmp/master/' X_train = pd.read_csv(INPUT_DIR+'X_train.csv') y_train = np.ravel(pd.read_json(INPUT_DIR+'y_train.json').values) thundergbm.TGBMRegressor().fit(X_train, y_train)

Kurt-Liuhf commented 4 years ago

Hi @ArtyomSalnikov , thanks for your feedback. We have fixed the bugs and you can now reinstall thundergbm via pip.