机器学习模型：我应该定义一个类或函数

Question

我使用Python编写自己的机器学习模型进行练习。我有两种方法可以解决它：

1. 写一个函数，例如def logisticRegression(...):

2. 写一个类，例如class logisticRegression:

显然，两者都有效。但哪个更像Pythonic？

Answer 1

对我而言，使用类方法更有意义，因为您可以将模型保存为类的实例，并使用train()作为类函数来初始化模型和predict()多次使用模型而不必重新训练的方法。

查看scikit-learn class for Logistic Regression，它非常有意义并且非常直观。