Python+Tensorflow学习(二)——初试keras
学习视频链接:(强推)TensorFlow官方入门实操课程
源码如下:
# -*- coding = utf-8 -*-
# @Time : 2021/8/9 10:27
# @Author : 西兰花
# @File : tf01.py
# @Software : PyCharm
from tensorflow import keras # Keras是tensorflow中的一个高级API
import numpy as np
"""
例1:如何根据X得到Y?
x = -1, 0,1,2,3, 4
y = -3,-1, 1,3,5,7
经观察,x、y关系如下:
y = 2x - 1
"""
'''构建模型'''
# layers表示神经元层数,units表示每层神经元中的个数,input表示输入值个数,如题中只有x是输入值
# optimizer表示指定的优化方式,loss表示指定的损失函数,
model = keras.Sequential([keras.layers.Dense(units=1, input_shape=[1])])
model.compile(optimizer='sgd', loss='mean_squared_error')
'''准备训练数据'''
xs &