实战Kaggle比赛：预测房价

:label:sec_kaggle_house

之前几节我们学习了一些训练深度网络的基本工具和网络正则化的技术（如权重衰减、暂退法等）。 本节我们将通过Kaggle比赛，将所学知识付诸实践。 Kaggle的房价预测比赛是一个很好的起点。 此数据集由Bart de Cock于2011年收集 :cite:De-Cock.2011， 涵盖了2006-2010年期间亚利桑那州埃姆斯市的房价。 这个数据集是相当通用的，不会需要使用复杂模型架构。 它比哈里森和鲁宾菲尔德的波士顿房价 数据集要大得多，也有更多的特征。

本节我们将详细介绍数据预处理、模型设计和超参数选择。 通过亲身实践，你将获得一手经验，这些经验将有益数据科学家的职业成长。

下载和缓存数据集

在整本书中，我们将下载不同的数据集，并训练和测试模型。 这里我们(实现几个函数来方便下载数据)。 首先，我们建立字典DATA_HUB， 它可以将数据集名称的字符串映射到数据集相关的二元组上， 这个二元组包含数据集的url和验证文件完整性的sha-1密钥。 所有类似的数据集都托管在地址为DATA_URL的站点上。

```{.python .input}

@tab all

import os import requests import zipfile import tarfile import hashlib

@save

DATA_HUB = dict() DATA_URL = ‘http://d2l-data.s3-accelerate.amazonaws.com/‘

  2. 下面的`download`函数用来下载数据集，
  3. 将数据集缓存在本地目录（默认情况下为`../data`）中，
  4. 并返回下载文件的名称。
  5. 如果缓存目录中已经存在此数据集文件，并且其sha-1与存储在`DATA_HUB`中的相匹配，
  6. 我们将使用缓存的文件，以避免重复的下载。
  8. ```{.python .input}
  9. #@tab all
  10. def download(name, cache_dir=os.path.join('..', 'data')):  #@save
  11.     """下载一个DATA_HUB中的文件，返回本地文件名"""
  12.     assert name in DATA_HUB, f"{name} 不存在于 {DATA_HUB}"
  13.     url, sha1_hash = DATA_HUB[name]
  14.     os.makedirs(cache_dir, exist_ok=True)
  15.     fname = os.path.join(cache_dir, url.split('/')[-1])
  16.     if os.path.exists(fname):
  17.         sha1 = hashlib.sha1()
  18.         with open(fname, 'rb') as f:
  19.             while True:
  20.                 data = f.read(1048576)
  21.                 if not data:
  22.                     break
  23.                 sha1.update(data)
  24.         if sha1.hexdigest() == sha1_hash:
  25.             return fname  # 命中缓存
  26.     print(f'正在从{url}下载{fname}...')
  27.     r = requests.get(url, stream=True, verify=True)
  28.     with open(fname, 'wb') as f:
  29.         f.write(r.content)
  30.     return fname

我们还需实现两个实用函数： 一个将下载并解压缩一个zip或tar文件， 另一个是将本书中使用的所有数据集从DATA_HUB下载到缓存目录中。

```{.python .input}

@tab all

def download_extract(name, folder=None): #@save “””下载并解压zip/tar文件””” fname = download(name) base_dir = os.path.dirname(fname) data_dir, ext = os.path.splitext(fname) if ext == ‘.zip’: fp = zipfile.ZipFile(fname, ‘r’) elif ext in (‘.tar’, ‘.gz’): fp = tarfile.open(fname, ‘r’) else: assert False, ‘只有zip/tar文件可以被解压缩’ fp.extractall(base_dir) return os.path.join(base_dir, folder) if folder else data_dir

def download_all(): #@save “””下载DATA_HUB中的所有文件””” for name in DATA_HUB: download(name)

  2. ## Kaggle
  4. [Kaggle](https://www.kaggle.com)是一个当今流行举办机器学习比赛的平台，
  5. 每场比赛都以至少一个数据集为中心。
  6. 许多比赛有赞助方，他们为获胜的解决方案提供奖金。
  7. 该平台帮助用户通过论坛和共享代码进行互动，促进协作和竞争。
  8. 虽然排行榜的追逐往往令人失去理智：
  9. 有些研究人员短视地专注于预处理步骤，而不是考虑基础性问题。
  10. 但一个客观的平台有巨大的价值：该平台促进了竞争方法之间的直接定量比较，以及代码共享。
  11. 这便于每个人都可以学习哪些方法起作用，哪些没有起作用。
  12. 如果你想参加Kaggle比赛，你首先需要注册一个账户（见 :numref:`fig_kaggle`）。
  14. ![Kaggle网站](/uploads/projects/d2l-ai-CN/img/kaggle.png)
  15. :width:`400px`
  16. :label:`fig_kaggle`
  18. 在房价预测比赛页面（如 :numref:`fig_house_pricing` 所示），
  19. 你在"Data"选项卡下可以找到数据集。
  20. 你可以通过下面的网址提交预测，并查看排名：
  22. >https://www.kaggle.com/c/house-prices-advanced-regression-techniques
  24. ![房价预测比赛页面](/uploads/projects/d2l-ai-CN/img/house-pricing.png)
  25. :width:`400px`
  26. :label:`fig_house_pricing`
  28. ## 访问和读取数据集
  30. 注意，竞赛数据分为训练集和测试集。
  31. 每条记录都包括房屋的属性值和属性，如街道类型、施工年份、屋顶类型、地下室状况等。
  32. 这些特征由各种数据类型组成。
  33. 例如，建筑年份由整数表示，屋顶类型由离散类别表示，其他特征由浮点数表示。
  34. 这就是现实让事情变得复杂的地方：例如，一些数据完全丢失了，缺失值被简单地标记为“NA”。
  35. 每套房子的价格只出现在训练集中（毕竟这是一场比赛）。
  36. 我们将希望划分训练集以创建验证集，但是在将预测结果上传到Kaggle之后，
  37. 我们只能在官方测试集中评估我们的模型。
  38. 在 :numref:`fig_house_pricing` 中，"Data"选项卡有下载数据的链接。
  40. 开始之前，我们将[**使用`pandas`读入并处理数据**]，
  41. 这是我们在 :numref:`sec_pandas`中引入的。
  42. 因此，在继续操作之前，你需要确保已安装`pandas`。
  43. 幸运的是，如果你正在用Jupyter阅读该书，你可以在不离开笔记本的情况下安装`pandas`。
  45. ```{.python .input}
  46. # 如果你没有安装pandas，请取消下一行的注释
  47. # !pip install pandas
  49. %matplotlib inline
  50. from d2l import mxnet as d2l
  51. from mxnet import gluon, autograd, init, np, npx
  52. from mxnet.gluon import nn
  53. import pandas as pd
  54. npx.set_np()

```{.python .input}

@tab pytorch

如果你没有安装pandas，请取消下一行的注释

!pip install pandas

%matplotlib inline from d2l import torch as d2l import torch from torch import nn import pandas as pd import numpy as np

  2. ```{.python .input}
  3. #@tab tensorflow
  4. # 如果你没有安装pandas，请取消下一行的注释
  5. # !pip install pandas
  7. %matplotlib inline
  8. from d2l import tensorflow as d2l
  9. import tensorflow as tf
  10. import pandas as pd
  11. import numpy as np

为方便起见，我们可以使用上面定义的脚本下载并缓存Kaggle房屋数据集。

```{.python .input}

@tab all

DATA_HUB[‘kaggle_house_train’] = ( #@save DATA_URL + ‘kaggle_house_pred_train.csv’, ‘585e9cc93e70b39160e7921475f9bcd7d31219ce’)

DATA_HUB[‘kaggle_house_test’] = ( #@save DATA_URL + ‘kaggle_house_pred_test.csv’, ‘fa19780a7b011d9b009e8bff8e99922a8ee2eb90’)

  2. 我们使用`pandas`分别加载包含训练数据和测试数据的两个CSV文件。
  4. ```{.python .input}
  5. #@tab all
  6. train_data = pd.read_csv(download('kaggle_house_train'))
  7. test_data = pd.read_csv(download('kaggle_house_test'))

训练数据集包括1460个样本，每个样本80个特征和1个标签， 而测试数据集包含1459个样本，每个样本80个特征。

```{.python .input}

@tab all

print(train_data.shape) print(test_data.shape)

  2. 让我们看看[**前四个和最后两个特征，以及相应标签**]（房价）。
  4. ```{.python .input}
  5. #@tab all
  6. print(train_data.iloc[0:4, [0, 1, 2, 3, -3, -2, -1]])

我们可以看到，(在每个样本中，第一个特征是ID，) 这有助于模型识别每个训练样本。 虽然这很方便，但它不携带任何用于预测的信息。 因此，在将数据提供给模型之前，(我们将其从数据集中删除)。

```{.python .input}

@tab all

all_features = pd.concat((train_data.iloc[:, 1:-1], test_data.iloc[:, 1:]))

  2. ## 数据预处理
  4. 如上所述，我们有各种各样的数据类型。
  5. 在开始建模之前，我们需要对数据进行预处理。
  6. 首先，我们[**将所有缺失的值替换为相应特征的平均值。**]然后，为了将所有特征放在一个共同的尺度上，
  7. 我们(**通过将特征重新缩放到零均值和单位方差来标准化数据**)：
  9. $$x \leftarrow \frac{x - \mu}{\sigma},$$
  11. 其中$\mu$和$\sigma$分别表示均值和标准差。
  12. 现在，这些特征具有零均值和单位方差，即 $E[\frac{x-\mu}{\sigma}] = \frac{\mu - \mu}{\sigma} = 0$和$E[(x-\mu)^2] = (\sigma^2 + \mu^2) - 2\mu^2+\mu^2 = \sigma^2$。
  13. 直观地说，我们标准化数据有两个原因：
  14. 首先，它方便优化。
  15. 其次，因为我们不知道哪些特征是相关的，
  16. 所以我们不想让惩罚分配给一个特征的系数比分配给其他任何特征的系数更大。
  18. ```{.python .input}
  19. #@tab all
  20. # 若无法获得测试数据，则可根据训练数据计算均值和标准差
  21. numeric_features = all_features.dtypes[all_features.dtypes != 'object'].index
  22. all_features[numeric_features] = all_features[numeric_features].apply(
  23.     lambda x: (x - x.mean()) / (x.std()))
  24. # 在标准化数据之后，所有均值消失，因此我们可以将缺失值设置为0
  25. all_features[numeric_features] = all_features[numeric_features].fillna(0)

接下来，我们[处理离散值。] 这包括诸如“MSZoning”之类的特征。 (我们用独热编码替换它们)， 方法与前面将多类别标签转换为向量的方式相同 （请参见 :numref:subsec_classification-problem）。 例如，“MSZoning”包含值“RL”和“Rm”。 我们将创建两个新的指示器特征“MSZoning_RL”和“MSZoning_RM”，其值为0或1。 根据独热编码，如果“MSZoning”的原始值为“RL”， 则：“MSZoning_RL”为1，“MSZoning_RM”为0。 pandas软件包会自动为我们实现这一点。

```{.python .input}

@tab all

“Dummy_na=True”将“na”（缺失值）视为有效的特征值，并为其创建指示符特征

all_features = pd.get_dummies(all_features, dummy_na=True) all_features.shape

  2. 你可以看到，此转换会将特征的总数量从79个增加到331个。
  3. 最后，通过`values`属性，我们可以
  4. [**从`pandas`格式中提取NumPy格式，并将其转换为张量表示**]用于训练。
  6. ```{.python .input}
  7. #@tab all
  8. n_train = train_data.shape[0]
  9. train_features = d2l.tensor(all_features[:n_train].values, dtype=d2l.float32)
  10. test_features = d2l.tensor(all_features[n_train:].values, dtype=d2l.float32)
  11. train_labels = d2l.tensor(
  12.     train_data.SalePrice.values.reshape(-1, 1), dtype=d2l.float32)

[训练]

首先，我们训练一个带有损失平方的线性模型。 显然线性模型很难让我们在竞赛中获胜，但线性模型提供了一种健全性检查， 以查看数据中是否存在有意义的信息。 如果我们在这里不能做得比随机猜测更好，那么我们很可能存在数据处理错误。 如果一切顺利，线性模型将作为基线（baseline）模型， 让我们直观地知道最好的模型有超出简单的模型多少。

```{.python .input} loss = gluon.loss.L2Loss()

def get_net(): net = nn.Sequential() net.add(nn.Dense(1)) net.initialize() return net

  2. ```{.python .input}
  3. #@tab pytorch
  4. loss = nn.MSELoss()
  5. in_features = train_features.shape[1]
  7. def get_net():
  8.     net = nn.Sequential(nn.Linear(in_features,1))
  9.     return net

```{.python .input}

@tab tensorflow

loss = tf.keras.losses.MeanSquaredError()

def get_net(): net = tf.keras.models.Sequential() net.add(tf.keras.layers.Dense( 1, kernel_regularizer=tf.keras.regularizers.l2(weight_decay))) return net

  2. 房价就像股票价格一样，我们关心的是相对数量，而不是绝对数量。
  3. 因此，[**我们更关心相对误差$\frac{y - \hat{y}}{y}$，**]
  4. 而不是绝对误差$y - \hat{y}$。
  5. 例如，如果我们在俄亥俄州农村地区估计一栋房子的价格时，
  6. 假设我们的预测偏差了10万美元，
  7. 然而那里一栋典型的房子的价值是12.5万美元，
  8. 那么模型可能做得很糟糕。
  9. 另一方面，如果我们在加州豪宅区的预测出现同样的10万美元的偏差，
  10. （在那里，房价中位数超过400万美元）
  11. 这可能是一个不错的预测。
  13. (**解决这个问题的一种方法是用价格预测的对数来衡量差异**)。
  14. 事实上，这也是比赛中官方用来评价提交质量的误差指标。
  15. 即将$\delta$ for $|\log y - \log \hat{y}| \leq \delta$
  16. 转换为$e^{-\delta} \leq \frac{\hat{y}}{y} \leq e^\delta$。
  17. 这使得预测价格的对数与真实标签价格的对数之间出现以下均方根误差：
  19. $$\sqrt{\frac{1}{n}\sum_{i=1}^n\left(\log y_i -\log \hat{y}_i\right)^2}.$$
  21. ```{.python .input}
  22. def log_rmse(net, features, labels):
  23.     # 为了在取对数时进一步稳定该值，将小于1的值设置为1
  24.     clipped_preds = np.clip(net(features), 1, float('inf'))
  25.     return np.sqrt(2 * loss(np.log(clipped_preds), np.log(labels)).mean())

```{.python .input}

@tab pytorch

def log_rmse(net, features, labels):

  1. # 为了在取对数时进一步稳定该值，将小于1的值设置为1
  2. clipped_preds = torch.clamp(net(features), 1, float('inf'))
  3. rmse = torch.sqrt(loss(torch.log(clipped_preds),
  4.                        torch.log(labels)))
  5. return rmse.item()
  2. ```{.python .input}
  3. #@tab tensorflow
  4. def log_rmse(y_true, y_pred):
  5.     # 为了在取对数时进一步稳定该值，将小于1的值设置为1
  6.     clipped_preds = tf.clip_by_value(y_pred, 1, float('inf'))
  7.     return tf.sqrt(tf.reduce_mean(loss(
  8.         tf.math.log(y_true), tf.math.log(clipped_preds))))

与前面的部分不同，[我们的训练函数将借助Adam优化器] （我们将在后面章节更详细地描述它）。 Adam优化器的主要吸引力在于它对初始学习率不那么敏感。

```{.python .input} def train(net, train_features, train_labels, test_features, test_labels, num_epochs, learning_rate, weight_decay, batch_size): train_ls, test_ls = [], [] train_iter = d2l.load_array((train_features, train_labels), batch_size)

  1. # 这里使用的是Adam优化算法
  2. trainer = gluon.Trainer(net.collect_params(), 'adam', {
  3.     'learning_rate': learning_rate, 'wd': weight_decay})
  4. for epoch in range(num_epochs):
  5.     for X, y in train_iter:
  6.         with autograd.record():
  7.             l = loss(net(X), y)
  8.         l.backward()
  9.         trainer.step(batch_size)
  10.     train_ls.append(log_rmse(net, train_features, train_labels))
  11.     if test_labels is not None:
  12.         test_ls.append(log_rmse(net, test_features, test_labels))
  13. return train_ls, test_ls
  2. ```{.python .input}
  3. #@tab pytorch
  4. def train(net, train_features, train_labels, test_features, test_labels,
  5.           num_epochs, learning_rate, weight_decay, batch_size):
  6.     train_ls, test_ls = [], []
  7.     train_iter = d2l.load_array((train_features, train_labels), batch_size)
  8.     # 这里使用的是Adam优化算法
  9.     optimizer = torch.optim.Adam(net.parameters(),
  10.                                  lr = learning_rate,
  11.                                  weight_decay = weight_decay)
  12.     for epoch in range(num_epochs):
  13.         for X, y in train_iter:
  14.             optimizer.zero_grad()
  15.             l = loss(net(X), y)
  16.             l.backward()
  17.             optimizer.step()
  18.         train_ls.append(log_rmse(net, train_features, train_labels))
  19.         if test_labels is not None:
  20.             test_ls.append(log_rmse(net, test_features, test_labels))
  21.     return train_ls, test_ls

```{.python .input}

@tab tensorflow

def train(net, train_features, train_labels, test_features, test_labels, num_epochs, learning_rate, weight_decay, batch_size): train_ls, test_ls = [], [] train_iter = d2l.load_array((train_features, train_labels), batch_size)

  1. # 这里使用的是Adam优化算法
  2. optimizer = tf.keras.optimizers.Adam(learning_rate)
  3. net.compile(loss=loss, optimizer=optimizer)
  4. for epoch in range(num_epochs):
  5.     for X, y in train_iter:
  6.         with tf.GradientTape() as tape:
  7.             y_hat = net(X)
  8.             l = loss(y, y_hat)
  9.         params = net.trainable_variables
  10.         grads = tape.gradient(l, params)
  11.         optimizer.apply_gradients(zip(grads, params))
  12.     train_ls.append(log_rmse(train_labels, net(train_features)))
  13.     if test_labels is not None:
  14.         test_ls.append(log_rmse(test_labels, net(test_features)))
  15. return train_ls, test_ls
  2. ## $K$折交叉验证
  4. 你可能还记得，我们在讨论模型选择的部分（ :numref:`sec_model_selection`）
  5. 中介绍了[**K折交叉验证**]，
  6. 它有助于模型选择和超参数调整。
  7. 我们首先需要定义一个函数，在$K$折交叉验证过程中返回第$i$折的数据。
  8. 具体地说，它选择第$i$个切片作为验证数据，其余部分作为训练数据。
  9. 注意，这并不是处理数据的最有效方法，如果我们的数据集大得多，会有其他解决办法。
  11. ```{.python .input}
  12. #@tab all
  13. def get_k_fold_data(k, i, X, y):
  14.     assert k > 1
  15.     fold_size = X.shape[0] // k
  16.     X_train, y_train = None, None
  17.     for j in range(k):
  18.         idx = slice(j * fold_size, (j + 1) * fold_size)
  19.         X_part, y_part = X[idx, :], y[idx]
  20.         if j == i:
  21.             X_valid, y_valid = X_part, y_part
  22.         elif X_train is None:
  23.             X_train, y_train = X_part, y_part
  24.         else:
  25.             X_train = d2l.concat([X_train, X_part], 0)
  26.             y_train = d2l.concat([y_train, y_part], 0)
  27.     return X_train, y_train, X_valid, y_valid

当我们在$K$折交叉验证中训练$K$次后，[返回训练和验证误差的平均值]。

```{.python .input}

@tab all

def k_fold(k, X_train, y_train, num_epochs, learning_rate, weight_decay, batch_size): train_l_sum, valid_l_sum = 0, 0 for i in range(k): data = get_k_fold_data(k, i, X_train, y_train) net = get_net() train_ls, valid_ls = train(net, *data, num_epochs, learning_rate, weight_decay, batch_size) train_l_sum += train_ls[-1] valid_l_sum += valid_ls[-1] if i == 0: d2l.plot(list(range(1, num_epochs + 1)), [train_ls, valid_ls], xlabel=’epoch’, ylabel=’rmse’, xlim=[1, num_epochs], legend=[‘train’, ‘valid’], yscale=’log’) print(f’折{i + 1}，训练log rmse{float(train_ls[-1]):f}, ‘ f’验证log rmse{float(valid_ls[-1]):f}’) return train_l_sum / k, valid_l_sum / k

  2. ## [**模型选择**]
  4. 在本例中，我们选择了一组未调优的超参数，并将其留给读者来改进模型。
  5. 找到一组调优的超参数可能需要时间，这取决于一个人优化了多少变量。
  6. 有了足够大的数据集和合理设置的超参数，$K$折交叉验证往往对多次测试具有相当的稳定性。
  7. 然而，如果我们尝试了不合理的超参数，我们可能会发现验证效果不再代表真正的误差。
  9. ```{.python .input}
  10. #@tab all
  11. k, num_epochs, lr, weight_decay, batch_size = 5, 100, 5, 0, 64
  12. train_l, valid_l = k_fold(k, train_features, train_labels, num_epochs, lr,
  13.                           weight_decay, batch_size)
  14. print(f'{k}-折验证: 平均训练log rmse: {float(train_l):f}, '
  15.       f'平均验证log rmse: {float(valid_l):f}')

请注意，有时一组超参数的训练误差可能非常低，但$K$折交叉验证的误差要高得多， 这表明模型过拟合了。 在整个训练过程中，你将希望监控训练误差和验证误差这两个数字。 较少的过拟合可能表明现有数据可以支撑一个更强大的模型， 较大的过拟合可能意味着我们可以通过正则化技术来获益。

[提交你的Kaggle预测]

既然我们知道应该选择什么样的超参数， 我们不妨使用所有数据对其进行训练 （而不是仅使用交叉验证中使用的$1-1/K$的数据）。 然后，我们通过这种方式获得的模型可以应用于测试集。 将预测保存在CSV文件中可以简化将结果上传到Kaggle的过程。

```{.python .input}

@tab all

def trainand_pred(train_features, test_features, train_labels, test_data, num_epochs, lr, weight_decay, batch_size): net = get_net() train_ls, = train(net, train_features, train_labels, None, None, num_epochs, lr, weight_decay, batch_size) d2l.plot(np.arange(1, num_epochs + 1), [train_ls], xlabel=’epoch’, ylabel=’log rmse’, xlim=[1, num_epochs], yscale=’log’) print(f’训练log rmse：{float(train_ls[-1]):f}’)

  1. # 将网络应用于测试集。
  2. preds = d2l.numpy(net(test_features))
  3. # 将其重新格式化以导出到Kaggle
  4. test_data['SalePrice'] = pd.Series(preds.reshape(1, -1)[0])
  5. submission = pd.concat([test_data['Id'], test_data['SalePrice']], axis=1)
  6. submission.to_csv('submission.csv', index=False)
  2. 如果测试集上的预测与$K$倍交叉验证过程中的预测相似，
  3. 那就是时候把它们上传到Kaggle了。
  4. 下面的代码将生成一个名为`submission.csv`的文件。
  6. ```{.python .input}
  7. #@tab all
  8. train_and_pred(train_features, test_features, train_labels, test_data,
  9.                num_epochs, lr, weight_decay, batch_size)

接下来，如 :numref:fig_kaggle_submit2中所示， 我们可以提交预测到Kaggle上，并查看在测试集上的预测与实际房价（标签）的比较情况。 步骤非常简单：

  • 登录Kaggle网站，访问房价预测竞赛页面。
  • 点击“Submit Predictions”或“Late Submission”按钮（在撰写本文时，该按钮位于右侧）。
  • 点击页面底部虚线框中的“Upload Submission File”按钮，选择你要上传的预测文件。
  • 点击页面底部的“Make Submission”按钮，即可查看你的结果。

向Kaggle提交数据 :width:400px :label:fig_kaggle_submit2

小结

  • 真实数据通常混合了不同的数据类型，需要进行预处理。
  • 常用的预处理方法：将实值数据重新缩放为零均值和单位方法；用均值替换缺失值。
  • 将类别特征转化为指标特征，可以使我们把这个特征当作一个独热向量来对待。
  • 我们可以使用$K$折交叉验证来选择模型并调整超参数。
  • 对数对于相对误差很有用。

练习

  1. 把你的预测提交给Kaggle，它有多好？
  2. 你能通过直接最小化价格的对数来改进你的模型吗？如果你试图预测价格的对数而不是价格，会发生什么？
  3. 用平均值替换缺失值总是好主意吗？提示：你能构造一个不随机丢失值的情况吗？
  4. 通过$K$折交叉验证调整超参数，从而提高Kaggle的得分。
  5. 通过改进模型（例如，层、权重衰减和dropout）来提高分数。
  6. 如果我们没有像本节所做的那样标准化连续的数值特征，会发生什么？

:begin_tab:mxnet Discussions :end_tab:

:begin_tab:pytorch Discussions :end_tab:

:begin_tab:tensorflow Discussions :end_tab: