期中考預測機率問題 - Cupoy

2019/10/27 下午 07:16

機器學習共學討論版

吳明宗

觀看數：17

回答數：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.]]

回答列表

• 2019/10/29 下午 10:57
  張維元 (WeiYuan)
  贊同數：0
  不贊同數：0
  留言數：1

  你這邊的 predict_proba 兩個值，是指 Y 欄位分別為兩種數值的機率。題目的要求應該只要其中一種而已。