初学tensorflow2,奈何不想去找样本,干脆自己生成一组。
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
path='E:/Desktop/python_code/Machine Learning/TensorFlow/dataset/'
x=np.random.uniform(-100,100,(100,))
#x=np.random.randn(100)
# 一开始看别人的博客写成了(100,1)
#于是报错x and y must be the same size
#但我打印x y的数量都是100
#后来通过打印x的ndim,发现为2 ,y的ndim为1。
#那看来是二维数组 不能跟一维数组绘制图表
c=np.random.uniform(-300,300,(100,))
# print(x.shape,x.ndim)
# print(c.shape,c.ndim)
y=10*x+c
df=pd.DataFrame({'x':x,'y':y})
df.to_csv(path+'liner.csv')
# print(len(x))
# print(len(c))
# print(len(y))
plt.scatter(x,y)
plt.show()
训练之后
发生异常: ValueError
x and y must be the same size
见代码注释