Question

我试图将gpu xgboost与以下代码结合使用，以使用xgboost创建多类模型。但是代码应该在训练开始后立即终止，并且会无任何错误消息地静默地执行，即只是静默退出并返回bash提示符。（上午Ubuntu 14.04 , xgboost 0.8 and python 3）

我认为这不是cuda或xgb版本的问题，因为我可以运行xgb sample很好。

问题可能是什么？感谢您的帮助。

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")])

带GPU的xgboost-静默终止

0 个答案: