oct*_*ian 5 python dataframe xgboost
我有以下代码:
from xgboost import XGBClassifier
print(df_train.shape)
print(df_train_labels.shape)
clf = clf.fit(df_train, df_train_labels, verbose=True)
print("after fit")
Run Code Online (Sandbox Code Playgroud)
这里df_train
和df_train_labels
是我从CSV读大熊猫。
上面的代码打印:
(1460, 7)
(1460,)
Run Code Online (Sandbox Code Playgroud)
但是,在 10 分钟内没有打印任何其他内容,这意味着代码卡住了,clf.fit
所以我假设算法不应该在此上花费很长时间。
如您所见,只有1460
示例,所以我假设算法不应该在此上花费很长时间。
此外,由于我通过了verbose=True
,我本来希望模型打印一些输出,但这并没有发生。
知道为什么没有打印输出以及为什么XGBClassifier
需要这么长时间吗?
use*_*583 -3
您应该首先根据您的情况初始化要使用的模型XGBClassifier()
。这样做的常见管道是:
from xgboost import XGBClassifier
model = XGBClassifier()
model.fit(df_train, df_train_labels)
y_pred = model.predict(df_test)
Run Code Online (Sandbox Code Playgroud)