tensorflow学习笔记

建立全连接网络:

import tensorflow as tf
import pandas as pd
import matplotlib.pyplot as plt

data = pd.read_csv('F:\\深度学习课程资料\\tensorflow\\数据集\\小型CSV数据\\Advertising.csv')
x = data.iloc[:,1:-1]
y = data.iloc[:,-1]
model = tf.keras.Sequential([tf.keras.layers.Dense(10,input_shape=(3,),
                             activation = 'relu'),
                             tf.keras.layers.Dense(1)]
                           )
model.compile(optimizer='adam',loss='mse')
model.fit(x,y,epochs = 1000)
test = data.iloc[:10,1:-1]
model.predict(test)

model = tf.keras.Sequential([tf.keras.layers.Dense(10,input_shape=(3,),
                             activation = 'relu'),
                             tf.keras.layers.Dense(1)]

这个语句是建立一层拥有10个神经元的隐含层和一层拥有一个神经元的输出层,Dense()中的参数第一个是输出的维数,第二个是输入的维度,第三个是激活函数。在建立层中只有第一层需要指明输入的维数,后面的只需要指明输出的层数即可

model.compile(optimizer='adam',loss='mse')

此语句是配置网络,另梯度下降优化器为adam,此优化器默认学习率为0.01,损失函数为均方差

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值