巴菲特的Alpha：利用机器学习量化『股票基本面』

这是机器学习发挥作用的地方。我们将建立一个机器学习分类器来判断一只股票是买入、卖出还是持有。为了确定一个股票或公司是否属于这三种类型之一，我们将看一下每个公司的季度报告。这些季度报告包含了必要的财务信息，我们需要用基本面分析的方法来训练我们的机器学习分类器。

1、基础数据——季度数据

为了训练分类器模型，我们需要收集大量的财务季度报告。国外数据从这里：

http://www.stockpup.com/data/

国内推荐Wind，因为比较全，财务量化因子也是PIT的。如果没有钱买他们的终端，可以用WindQuant万矿，这个是免费的，数据和Wind一致：

www.windquant.com

2、整理数据

DataFrame显示如下：

我们可以观察到所有不同的列和对应于每个列的日期。Stockpup还提供了价格信息（最高、最低等），这对我们判断股票是买进、持有还是卖出非常重要。

我们有很多种方法来决定股票是否值得投资。如果在过去的三个季度中，资产增加了，负债减少了，我们是否可以将其归类为买入？或者，如果股价上涨，长期债务下降？无论怎样，我们有许多选择来确定股票的类别。简单地说，这是基本面分析的基础。

在不了解基本面分析的情况下，我们如何从根本上分析这些季度报告？既然我们不是巴菲特那样的基本面专家，就让我们用自己的基本面分析方法来简化一下吧：

• 根据所选季度的季度报告，观察从上一个报告到当前所选报告的值的变化。
• 然后，观察下一个季度报告中的价格值，看看是否有显著的价格跳跃。
• 最后，使用当前报告包含过去报告的变化和未来报告的价格行为，确定是买进、持有还是卖出。

从本质上讲，我们正在检测上一季度到当前季度的变化是否会影响未来的价格。我们将根据上一份报告来判断每一份季度报告的表现，然后观察未来的价格走势。如下图所示：

我们将在每个季度报告中应用这种分析方法来创建新的基础数据。如果一只股票本季度值得投资，这种方法将进行分类。显然，我们不能在最初的 QR 或最近的 QR 上使用这种方法，因为分析的要求过去和未来的季度报告。

1、创建类标签

为了对每一份季度报告进行分类，我们将尽量不把事情复杂化。如果下个季度价格大幅上涨，那就是买入。如果下跌，那就是卖出。如果两者都不是，那就继续持有。

以下是我们对每个季度报告的具体类别要求：

• 买：下一季度，高价和低价的涨幅均在3% 以上

• 卖：下一季度，高价和低价跌幅达 -3% 或以上

• 持有：如果两者都没有发生

我们还可以考虑其他可能的结果，但为了简单起见，我们保持这些条件不变。我们不知道下个季度最新季度报告的价格高低，因为这是对未来的预测，而这是不可能的。

2、季度报告中的基本面数据

如前所述，对于我们的基本面数据，我们将观察两个QRs来创建新值。以前的QR到当前或当前QR的变化将以百分比变化来度量，而不是它们的实际值。例如：

• 假设上一季度，股票的价值是1000美元。而在本季度，股价目前为1100美元；增加了10%。现在，我们用当前QR的10%来替换1100的股票价值。

我们对每个QR都这样做（不包括第一个QR，因为我们无法将它与不存在的东西进行比较）。现在每个QR都有每个基本值的变化百分比。

3、编码

现在我们已经用百分比变化衡量了QRs，并标记为买入、持有或卖出，下面通过代码实现：

a. 导入数据集

pickle文件是一个DataFrames字典，其中包含 stockpup 网站上每个股票代码 / 公司的 QRs。

b. 辅助函数

c. 每个股票代码来转换数据

有几点需要注意：

1、为了在以后训练我们的分类模型，所有来自DFs字典的 DataFrames 被合并为一个。

2、索引被重置，因为不再需要日期。每一行或每一个 QR 现在都包含了过去和未来 QRs 的信息，所以日期对于模型不再重要

3、删除与价格相关的特征或列，以防止数据泄漏。在正常的QRs中，这些特征不包括在内。

最后，我们导出最终的DF用于数据探索和训练分类模型。

接下来，为了更加熟悉我们的数据，我们必须执行一些简单的探索性数据分析。我们这样做是为了更好地理解数据，并确保在转换数据时没有遗留。

如大家所见，我们的数据存在一个类不平衡问题。这可能是一个问题，但我们不希望丢弃数据点，以便每个类都等于数量最少的类。尽管这可能是解决不平衡的可行方案，但是我们还有其他的选择可以探索。另一个选择是在验证我们的分类模型时使用不同的评估指标。当进入建模阶段时，我们将对此选项进行扩展。

数据相关性：

从我们所看到的，一些特征对确定类标签有影响。有些与股票/季度报告是否值得买进、卖出或持有几乎没有关系。既然我们知道QR中的一些特征在确定类标签时并不重要，那么我们可以从数据集中删除这些特征。

现在我们已经对我们的数据进行了一些研究，我们可以继续进行更多的研究，或者转移到特征工程中，或者更具体地说，是特征选择。特征工程是改变数据集以增强机器学习模型的过程。有许多选项可以用来设计数据集的特征。包括但不限于：

• 创建交互功能。两个截然不同的特征相互作用，从而创建一个全新的特征。
• 减少特征的数量。由于我们有30多个基本特征，我们可以消除不重要的特征以提高模型的性能。
• 使用领域知识。如果根据经济和金融研究，我们知道哪些特征对股票价格最重要，那么我们就可以给这些特征赋予权重。

1、特征选择

在本文中，我们将简单地删除不重要的特征。这样做，我们可以潜在地提高模型的准确性，并减少训练时间。我们将展示两种不同的特征选择方法。这样，我们可以比较由于我们方法的不同，模型的性能与略有不同的数据集会产生什么样的结果。

方法1：根据相关值选择前10个特征

我们选择与决策类标签相关的前10个特征：

• 取每个特征的绝对值（处理负相关性）。
• 按绝对值排序。
• 将数字特征分割为只包含前10个最相关的特征。
• 加入决策类标签，以便将它们包含在新的DF中。
• 将前10名的Correlation DF导出为pickle文件。

我们现在有了一个新的分类模型数据集。希望这个新的数据集能增强我们的分类模型，并加快训练时间。

方法2：基于树分类器选择前10个特征

下一种特征选择方法比我们刚才做的要复杂一些。通过使用sklearn，我们将实现一个基于决策树的分类器来确定哪些特征是最重要的。通过使用这个分类器，我们将能够确定哪些特征是最重要的分类QR或股票。

接下来，随着重要特征的排序，我们将其可视化，以确定哪些特征被确定为最重要的。 

根据forest.feature_importances_方法，以下是十个是最重要的特征：

• P/B Ratio
• Assets
• Shares split adjusted
• Shares
• P/E Ratio
• Asset turnover
• Liabilities
• Current Assets
• Cash at end of period
• Revenue

为了找到性能最佳的机器学习分类模型，我们将通过多个模型运行数据。每种模式都有其独特的优点和缺点。为了确保我们对数据使用的是最优模型，我们必须尝试每个分类器。

我们将使用的分类模型:

• AdaBoost
• 决策树
• DummyClassifier
• Gradient Boost
• KNN
• 逻辑回归
• 朴素贝叶斯
• 随机森林
• SVM
• XGBoost

有很多模型!它们中的每一个都将被训练和安装到我们的数据集，以确定最佳性能的模型。

1、DummyClassifier

为了度量每个模型的性能，需要将它们与基线模型进行比较。这就是我们的虚拟分类器的用武之地。虚拟分类器的功能类似于随机猜测每个QR的类标签。它通常是基于概率的，类标签是根据它们的样本大小来决定的。

以建立的虚拟分类器为基线模型，为每个模型设定一个性能标准。为了被认为是一个可行的模型，每个模型必须能够超越虚拟分类器。如果他们不能超越基线模型，那么模型也不会比随机猜测更好。

1、导入库并加载数据：

在导入数据时，我们可以选择需要哪个数据集。如果我们希望比较两个不同数据集的模型性能，这是很重要的。

2、缩放和训练，测试，分割数据：

使用的缩放方法（归一化）主要取决于大家，因为每个缩放器都有自己独特的优势。然后我们将数据分解为X和Y变量的训练集和测试集，我们将需要这些数据来拟合和评估我们的模型。

3、创建虚拟分类器：

我们通过实例化分类器来创建虚拟分类器，然后将其匹配到我们的训练集。接下来，我们使用测试集来制作预测标签，并与实际标签进行比较。最后，我们打印出一份包含我们需要的评估指标的分类报告：

由于我们处理的是一个不平衡的数据集，我们不能依赖准确性作为我们的模型的评估指标，这就是为什么它没有包括在分类报告中。准确性可能是一个重要的衡量标准，但是由于我们问题的本质和类别的不平衡，准确度将被忽略，而有利于准确率和召回率。

精确率是针对我们预测结果而言的：它表示的是预测为正的样本中有多少是真正的正样本。

召回率是针对我们原来的样本而言的：它表示的是样本中的正例有多少被预测正确了。

精确率>召回率

为什么我们选择精确率而不是召回率呢？当涉及到投资决策时，我们不希望分类器将卖出错误地归类为买入。在我们保守的投资策略中，我们宁愿错过投资机会（召回率），也不愿投资错误的股票（精确率）。很明显，两者之间有一个平衡，这就是F1-Score发挥作用的地方。但是，每个分类模型的总体判断将基于其相对于基线模型的精度评分的增加。

对于每个分类器，我们将可视化它们的混淆矩阵以获得更好的可解释性。混淆矩阵将显示被错误分类的QRs的数量。

随着混乱矩阵的可视化，以及基线模型的创建和评估，我们已经正式完成了我们的第一个分类模型。

要创建更多的模型，我们所要做的只是复制基线模型的代码，然后修改几行代码。下面是一个模型的代码（与DummyClassifier非常相似）：

正如我们所看到的，在代码中只有两到三行需要修改来构建每个模型。

我们需要改变的两三行是:

1、with open(“top10_df.pkl”, “rb”) as fp：只有在测试其他不同特征的数据集时才进行修改。

2、from sklearn.ensemble import AdaBoostClassifier：将这些导入更改为适当的库及其各自的分类器。

3、clf = AdaBoostClassifier()：将这个变量更改为适当的分类器。

其他一切都保持不变。

在调整模型以适应数据集时，我们可以保持模型不变，也可以通过使用网格搜索来提高模型的性能。通过使用这种方法，我们能够适当地根据我们的数据集调整每个分类模型的参数。

到目前为止，网格搜索将是最耗时的。不是因为我们的代码很多，而是因为每个参数组合的迭代数量。有些模型需要比其他模型更长的时间来进行网格搜索，但是整个过程仍然需要相当多的时间，因为我们将网格搜索我们拥有的每个模型。

网格搜索的代码对于每个模型基本上是相同的，我们只需要修改几行代码。

正如大家所看到的，需要编写的代码并不多，实际上这里有一些我们以前使用过的代码（分类报告和混淆矩阵）。

需要改变的是：

• params变量，它将是一个字典，字典的键对应于相应的分类器参数，字典的值是我们希望为每个迭代尝试的不同值。

• 当变量最初被更改以适应每个新的分类器时，clf早就应该被更改了。

1、网格搜索的参数

这些参数需要设置为适合我们自己的个人需求。正如我们所看到的，clf 和 params 已在上面处理。但对于其他参数：

• return_train_score设置为True，以便以后观察评估的结果。
• verbose设置为5，这样我们就可以看到每个迭代的进度和它们的分数。
• scoring设置为f1_macro，因为我们在确定最佳参数时优先考虑F1评分。

2、F1分数宏平均（macro-average)

F1分数优先于精确率的原因仅仅是分类的数量。当精确率被优先化时，精确率分数可能会显著增加，但被归类为某些类别的QRs的数量会大大减少。 在我们的示例中，分类器很少对 Buy 或 Sell 进行分类，但是许多分类都在 Hold 类中。 这是因为 Hold 类是大多数类。 我们不希望我们的分类器很少告诉我们什么时候投资，我们希望至少有一定数量的机会属于召回率得分。

为了在精确率和召回率之间取得平衡，我们会优先考虑F1的分数。我们之所以选择宏观平均而不是微观平均，是因为宏观忽视了阶级不平衡。持有类，作为占主导地位的类，当涉及到投资活动时，我们几乎什么都不知道，因为我们更想知道股票是买还是卖。总的来说，使用F1分数宏平均值将允许网格搜索在我们喜欢的条件下找到最佳参数。

现在我们有了数据、分类器代码、评估报告和网格搜索代码，我们可以将它们应用到每个分类模型中。

1、保存分类模型

一旦我们决定了哪个模型最适合我们的数据，我们就可以用正确调优的参数保存拟合的模型。看看下面的代码，看看如何保存一个模型（只有两行代码）：

这段代码将导出我们的模型以供其他文件使用。

2、表现最佳的模型

在我们的例子中，性能最好的模型来自XGBoost分类器，它使用了来自树分类数据集的前10个重要特征。该分类器的分类报告如下：

正如我们在这里看到的，与基线模型相比，在XGBoost模型中，Buy类增加了10%，Sell类增加了20%。这是对DummyClassifier模型的重大改进。在很多方面，我们基本上是在预测未来。如果我们的分类器能够以47%的准确率来决定一个股票是否值得投资，那么我们应该认为这是一个重大的突破！

假设我们想用分类器来预测一个新的QR。我们怎么才能做到呢？最好的选择是找到最近的QR和以前的QR的值。然后，观察它们之间的百分比变化，以便它们与我们自己的数据集的格式匹配。

新的数据可以从许多报告QRs的网站获得，我们可以通过网络搜集这些值，也可以手工输入。Stockpup也有最新的QRs，对于本文来说，我们将利用该网站的最新值。在前面的部分中，我们从每个库存中删除了最新的QR，因为没有未来的最高和最低价格用于我们的类标签。所以这些QRs仍然没有被标记，但是现在我们可以用分类器的预测来标记它们。

1、最新的QR和可视化辅助函数：

在处理新数据时，为了与配合分类器，我们必须对数据进行扩展，因为我们最初就是这样训练分类器的。必须对数据进行处理，以包含百分比修正、正确的特征列和与其他相关数据的缩放值。一旦我们有了所有这些，我们就可以对最新的QR进行分类。

2、分类最新的QR

在这个例子中，我们将使用来自AMD的最新QR：

根据我们的分类器，AMD有42.5%的机会持有。最主要的概率将决定分类器如何确定股票的名称。

根据季度报告来决定股票价值并不是什么新成就。对于大多数基础分析人员来说，我们使用的策略可能被认为过于简单或过时，但是对于学习机器学习分类来说，这已经足够了。我们可以添加更多的特征来改进模型，或者根据新的策略改变现有的特征。

我们甚至可以用神经网络做实验，看看它们是否能比我们现有的模型表现得更好。我们将在未来的文章中给大家讲解！

2020年第22篇文章

你点的每个“在看”，都是对我们最大的鼓励