ELI5作为熊猫DataFrame的explain_weights和explain_predictions

时间:2019-01-28 16:18:55

标签: keras scikit-learn

如何将ELI5 prodexplain_weights保存为pandas DataFrame?

以下示例玩具代码。

explain_prediction

3 个答案:

答案 0 :(得分:0)

假设您只想选择 X_test 数据框中的第一行,请更改以下内容:

explanation_pred = eli5.formatters.as_dataframe.explain_prediction_df(estimator = my_model,doc = X_test [0]

对此:

explanation_pred = eli5.formatters.as_dataframe.explain_prediction_df(estimator = my_model,doc = X_test.iloc [0]

对于 X_test 中的每个后续行,更改行号。例如,X_test.iloc [ 1 ],X_test.iloc [ 2 ],...,X_test.iloc [ n ]

问题是由您从 X_test 数据框中选择数据的方式引起的。您可以在Pandas中查看有关索引和selectind数据的完整文档:https://pandas.pydata.org/pandas-docs/stable/user_guide/indexing.html

答案 1 :(得分:-1)

尝试一下。它对我有用。

explanation_pred = eli5.explain_prediction_df(estimator=my_model, doc=X_test.iloc[0])

答案 2 :(得分:-2)

如果您仍在寻找答案。 您只需要替换以下代码行:

explanation_pred = eli5.formatters.as_dataframe.explain_prediction_df(estimator=my_model, 
doc=X_test[0])

收件人:

explanation_pred = eli5.explain_prediction_df(estimator=my_model, doc=X_test[0])