我想通过实现树解释器来解释RandomForestRegressor的结果。但是我收到一个断言错误:
`AssertionError Traceback (most recent call last)
<ipython-input-59-9966bd0a8cfd> in <module>
----> 1 assert(numpy.allclose(rf_prediction, rf_bias + np.sum(contributions, axis=1)))
AssertionError: `
代码如下:
`rf_prediction, rf_bias, rf_contributions = ti.predict(rf, test_features)
rf_prediction[0], rf_bias[0]
print(rf_bias + np.sum(contributions, axis=1))
assert(numpy.allclose(rf_prediction, rf_bias + np.sum(contributions, axis=1)))
assert(numpy.allclose(rf.predict(test_features), bias + np.sum(contributions, axis=1))`
真的无法在线找到有关此“断言错误”的任何指针吗?,谢谢您!
答案 0 :(得分:0)
您将同时使用带有预测和偏差的断言,并且对于给定的方法,它不会返回true。这样您会得到断言错误。
代替使用
numpy.allclose()
assert((rf.predict(test_features), bias + np.sum(contributions, axis=1))