自定义导入的 python 模块

Question

我有一个关于 python 模块的问题。我正在研究 IBM watson studio 的笔记本。我想将 KernelDensityEstimation 模型转储（部署）为 Web 服务。但是这个 KDE 模块没有 "predict" 功能。如果我想从 IBM studio 部署机器学习模型，我假设它们应该具有 "predict" 功能。所以现在我想将此功能添加到 KernelDensity 模块中。如何才能做到这一点？或者有什么方法可以部署这个模型。谢谢

Answer 1

您可以通过继承KDE 模块，通过您自己的自定义模块来实现这一点。您需要做的是创建一个模块并添加一个继承自 KernelDensity 的自定义 class，并在该 class 中添加您自己的方法，以便您可以访问 KernelDensity[=] 的所有 class 属性12=]

示例：

from sklearn.neighbors import KernelDensity

class MyKernelDensity(KernelDensity):
    def predict():
        # Your code goes here
        pass

定义这个之后，您应该始终从自定义模块中导入 MyKernelDensity 而不是导入 KernelDensity

即

from my_module import MyKernelDensity

density = MyKernelDensity(bandwidth=bandwidth, **kwargs)
density.predict()

我只提供了示例，您必须相应地重写。