2.5-缓解过拟合代码难点

参考视频：2.5-缓解过拟合_哔哩哔哩_bilibili

p29中前段代码：

# 导入所需模块
import tensorflow as tf
from matplotlib import pyplot as plt
import numpy as np
import pandas as pd

# 读入数据/标签 生成x_train y_train
df = pd.read_csv('dot.csv')
x_data = np.array(df[['x1', 'x2']])
y_data = np.array(df['y_c'])

x_train = np.vstack(x_data).reshape(-1, 2)
y_train = np.vstack(y_data).reshape(-1, 1)

使用了np.vstack().reshape(),其本身的目的是为了防止存在一维数组，把x_data和y_data转换为二维数组。

x_data本身已经是二维数组的了 以下为部分数据：

以下是x_train：

但是y_data本身是一维数组：

首先通过vstack将y_data堆叠成为二维数组，再使用reshape（-1,1）改变为任意行，一列的shape，以下为部分数据：