我在使用Python编写自己的机器学习模型以进行练习。我有两种方法可以选择:
-
编写一个函数,例如
def logisticRegression(...):
-
编写一个类,例如
class logisticRegression:
显然,这两种方法都可以工作。但哪一种更符合Python的风格呢?
回答:
对我来说,使用类的方法更有意义,因为你可以将模型保存为类的实例,并且可以将类的方法设置为类似于 train()
来初始化模型,以及 predict()
方法来多次使用你的模型,而无需重新训练它。
看看scikit-learn的Logistic Regression类,它非常合理且直观。