xgboost с графическим процессором - молча завершается - PullRequest
Новая
       7

xgboost с графическим процессором - молча завершается

0 голосов
/ 19 сентября 2018

Я пытаюсь использовать gpu xgboost со следующим кодом для создания мультиклассовой модели с xgboost.Но код завершается сразу после того, как должно начаться обучение, и делает это тихо без какого-либо сообщения об ошибке , то есть просто тихо завершает работу обратно в приглашение bash.(Я на Ubuntu 14.04 , xgboost 0.8 and python 3)

Я не думаю, что это проблема версии cuda или xgb, так как я смог запустить этот образец xgb отлично.

что моглопроблема будет?Спасибо за вашу помощь. 

def load_training_data():
    x_train = []
    y_train = []
    with open('./train_data.txt', 'rU') as f1:
        reader = csv.DictReader(f1)
        for row in reader:
            category = row["category"]
            if category.startswith('_'):continue
            metadata = row["metadata"]
            x_train.append(metadata)
            y_train.append(category)

    return x_train,y_train

print ("[+] loading data....")
X,y_ = load_training_data()
print ("[+] starting fit transform")
X_t = vectorizer.fit_transform(X)
le = LabelEncoder()
print ("[+] labels",set(y_))

y = le.fit_transform(y_)

# XGBoost API example
params = {'tree_method': 'gpu_hist', 'max_depth': 3, 'learning_rate': 0.1}
dtrain = xgb.DMatrix(X_t, y)
xgb.train(params, dtrain, evals=[(dtrain, "train")])
...