未定义“ TabularLIME” Azure DataBricks

问题描述

我刚刚开始研究Azure数据块。

在运行已经创建的Python笔记本时，我遇到了一些错误。

以下是代码：

import mmlspark
from mmlspark import *

lime = TabularLIME()\
  .setModel(randomForestModel)\
  .setPredictionCol("predict")\
  .setoutputCol("weights")\
  .setInputCol("feat")
lime_model = lime.fit(eng_train_pipe)
lime_model.save('dbfs:/mnt/aimodels/rf-eng-lag-lime')

出现此错误：

NameError                                 Traceback (most recent call last)
<command-2244283784192060> in <module>
      2 from mmlspark import *
      3 
----> 4 lime = TabularLIME()\
      5   .setModel(rfModel)\
      6   .setPredictionCol("prediction")\

NameError: name 'TabularLIME' is not defined

集群运行时为：6.5（包括Apache Spark 2.4.5，Scala 2.11）

解决方法

名称空间不正确。正确的名称空间是mmlspark.lime.TabularLIME.TabularLIME

from mmlspark.lime.TabularLIME import TabularLIME