logo
Loading...

期中考預測機率問題 - Cupoy

請問我這樣寫想要預測類別的機率,哪裡有錯???# 空值補平均值, 搭配最大最小化df = df.fi...

機率

期中考預測機率問題

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.]]