predict_proba用法
2020/04/12 02:33 下午
機器學習共學討論版
江柏勳
觀看數:3
回答數:3
收藏數:0
for col in app_train:
if app_train[col].dtype == 'object':
# 如果只有兩種值的類別型欄位
if len(list(app_train[col].unique())) <= 2:
# 就做 Label Encoder
le.fit(app_train[col])
app_train[col] = le.transform(app_train[col])
app_test[col] = le.transform(app_test[col])
想請問就是這邊可以用fit_transform()嗎?然後為何這邊不用le.fit(app_test[col])
2.ImportError: cannot import name 'Imputer'
這是我電腦的問題嗎?
# 請注意羅吉斯迴歸是分類預測 (會輸出 0 的機率, 與 1 的機率), 而我們只需要留下 1 的機率這排
log_reg_pred = log_reg.predict_proba(test)[:, 1]
可以這樣寫是本身函式的性質嗎?