解释随机森林,AssertionError

时间:2020-07-07 09:06:57

标签: jupyter-notebook random-forest interpreter

我想通过实现树解释器来解释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))`

真的无法在线找到有关此“断言错误”的任何指针吗?,谢谢您!

1 个答案:

答案 0 :(得分:0)

您将同时使用带有预测和偏差的断言,并且对于给定的方法,它不会返回true。这样您会得到断言错误。

代替使用

    numpy.allclose()
    assert((rf.predict(test_features), bias + np.sum(contributions, axis=1))