期中考預測機率問題 - Cupoy

2019/10/27 11:16 AM

機器學習新手論壇

吳明宗

觀看數：0

回答數：1

收藏數：0

機率

請問我這樣寫想要預測類別的機率,哪裡有錯???

# 空值補平均值, 搭配最大最小化

df = df.fillna(df.mean())

df_temp = MinMaxScaler().fit_transform(df)

train_X = df_temp[:train_num]

test_X = df[train_num:]

estimator = LogisticRegression()

estimator.fit(train_X, train_Y)

pred = estimator.predict_proba(test_X)

錯誤訊息:

Exception: Data must be 1-dimensional

pred的內容如下,不知道是predict_proba需要做調整嗎:

[[0. 1.]
[0. 1.]
[0. 1.]
[0. 1.]
[0. 1.]
[0. 1.]
[0. 1.]
[0. 1.]
[1. 0.]
[0. 1.]
[0. 1.]
[0. 1.]
[0. 1.]
[0. 1.]
[0. 1.]
[0. 1.]
[0. 1.]
[0. 1.]
[0. 1.]
[0. 1.]
[0. 1.]
[0. 1.]
[0. 1.]
[0. 1.]
[0. 1.]
[0. 1.]
[0. 1.]
[0. 1.]
[0. 1.]
[0. 1.]
[0. 1.]
[0. 1.]
[0. 1.]]