混淆矩陣：RecursionError

Question

我一直在嘗試復制用於繪制混淆矩陣的在線教程，但遇到了遞歸錯誤，嘗試重置遞歸限制，但錯誤仍然存​​在。 代碼如下：

log = LogisticRegression()
log.fit(x_train,y_train)
pred_log = log.predict(x_train)
confusion_matrix(y_train,pred_log)

我得到的錯誤是：

---------------------------------------------------------------------------
RecursionError                            Traceback (most recent call last)
<ipython-input-57-4b8fbe47e72d> in <module>
----> 1 (confusion_matrix(y_train,pred_log))

<ipython-input-48-92d5242f8580> in confusion_matrix(test_data, pred_data)
      1 def confusion_matrix(test_data,pred_data):
----> 2     c_mat = confusion_matrix(test_data,pred_data)
      3     return pd.DataFrame(c_mat)

... last 1 frames repeated, from the frame below ...

<ipython-input-48-92d5242f8580> in confusion_matrix(test_data, pred_data)
      1 def confusion_matrix(test_data,pred_data):
----> 2     c_mat = confusion_matrix(test_data,pred_data)
      3     return pd.DataFrame(c_mat)

RecursionError: maximum recursion depth exceeded

火車的形狀和測試數據如下

x_train.shape,y_train.shape,x_test.shape,y_test.shape 
# ((712, 7), (712,), (179, 7), (179,))

嘗試過： sys.setrecursionlimit(1500)
但是仍然沒有解決方案。

Answer 1

看起來您正在遞歸調用相同的函數。 嘗試更改外部函數名稱。

  1 def confusion_matrix(test_data,pred_data): ----> 2 c_mat = confusion_matrix(test_data,pred_data) 3 return pd.DataFrame(c_mat) 

至

def confusion_matrix_pd_convertor(test_data,pred_data):
    c_mat = confusion_matrix(test_data,pred_data)
    return pd.DataFrame(c_mat)

log = LogisticRegression()
log.fit(x_train,y_train)
pred_log = log.predict(x_train)
confusion_matrix_pd_convertor(y_train,pred_log)