Catboost python 特征重要性缺少 1 个必需的位置参数：'value'

Question

我正在尝试计算 python 文件中的特征重要性。我通过 Spark 提交运行这个 python 文件。由于我们的数据节点没有安装 catboost 库，我手动加载它们 python.

然后我加载模型文件并尝试计算特征重要性。我收到以下错误

abc = model.get_feature_importance(type=catBoost.EFstrType().FeatureImportance(), prettified=True, thread_count=-1, verbose=False)
TypeError: __call__() missing 1 required positional argument: 'value'

请看下面的代码，我是运行ning。

def getCatBoostLibraries():
    files = os.listdir(os.getcwd())
    with zipfile.ZipFile(os.path.abspath("catboost zip file path"), 'r') as zip_ref:
        zip_ref.extractall(os.getcwd())
    catboostFldrPath = os.path.abspath(os.path.join(os.getcwd(), "catboost"))
    sys.path.append(catboostFldrPath)
    configFile = os.path.abspath(os.path.join(catboostFldrPath, "__init__.py"))
    spec = importlib.util.spec_from_file_location("catboost", configFile)
    catBoost = importlib.util.module_from_spec(spec)
    spec.loader.exec_module(catBoost)
    return catBoost

def getFeatureImportance(modelPath):
    catBoost = getCatBoostLibraries()
    model = catBoost.CatBoostRegressor()
    model.load_model(modelPath)
    abc = model.get_feature_importance(type=catBoost.EFstrType().FeatureImportance(), prettified=True, thread_count=-1, verbose=False)
    importance_score_df = pd.DataFrame(abc, columns=['features', 'featr_imp'])
    return importance_score_df

Answer 1

我在下一行中删除大括号后它起作用了

abc = model.get_feature_importance(type=catBoost.EFstrType.FeatureImportance, prettified=True, thread_count=-1, verbose=False)

Catboost python 特征重要性缺少 1 个必需的位置参数：'value'

Catboost python feature importance missing 1 required positional argument: 'value'

python

catboost