Question

我有一个机器学习算法，它涉及一系列步骤，例如清理数据，准备训练数据等。每个步骤都存储在python类的单独方法中。我想知道构造我的类的最佳实践方法是什么，以便在类实例化时自动执行这些步骤。

这就是我所做的（代码是说明性的，但这种方法适用于真实的算法）。感觉有点笨重。有更优雅的方式吗？

 class Kaggle():
    """
    An algorithm
    """

    def __init__( self ):
        self.bar = 1

    def step_one( self, some_text_data ):
        self.bar = 1 ** 2
        # Do some data cleaning
        # return processed data

    def step_two( self ):
        foo = step_one(baz)
        # do some more processing

    def step_three( self ):
        bar = step_two()
        # output results

    def run( self ):
        self.step_one()
        self.step_two()
        self.step_three()

if __name__ == "__main__":
    kaggle = Kaggle()
    kaggle.run()

Answer 1

如果你的目标是让对象在课堂实例化时自动执行＆＃34;，只需将self.run()放在 init 中：

def __init__(self):
    self.bar = 1
    self.run()

Answer 2

将所有来电置于__init__方法中。这不是你想要达到的目标吗？您可以添加一个带有默认值的标志，这样您就可以不运行测试。

def __init__( self, runtests=True ):
    self.bar = 1
    if runtests:
        self.step_one()
        self.step_two()
        self.step_three()

在实例化时自动调用Python类方法的最佳方法

2 个答案: