Tensorflow2.0学习(一)Hello World!

TensorFlow2.0初体验吧。之前搞定了cuda环境,所以是用支持GPU的版本跑的,网上的教程大多是用tf1.x写的,所以想自己看着官网的文档来写一写2.0的教程。

安装十分简单:

    pip install tensorflow-gpu==2.0.0-beta1
   

新版本TensorFlow已经支持cuda10.0(谢天谢地)。

下面是官网给的示例代码:

import tensorflow as tf
mnist = tf.keras.datasets.mnist

(x_train, y_train),(x_test, y_test) = mnist.load_data()
x_train, x_test = x_train / 255.0, x_test / 255.0

model = tf.keras.models.Sequential([
  tf.keras.layers.Flatten(input_shape=(28, 28)),
  tf.keras.layers.Dense(512, activation=tf.nn.relu),
  tf.keras.layers.Dropout(0.2),
  tf.keras.layers.Dense(10, activation=tf.nn.softmax)
])
model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy'])

model.fit(x_train, y_train, epochs=5)
model.evaluate(x_test, y_test)

搭建一个程序共有五步:

1.创建Sequential模型;

2.添加所需要的神经层;在示例程序中,创建模型和添加神经层同步完成了,我们还可以用model.add()来添加神经层,并设置其功能,形状,激活函数,留存率等属性;

3使用.compile方法确定模型训练结构;有三个重要参数:

optimizer:此对象会指定训练过程。从 tf.train 模块向其传递优化器实例;
loss:要在优化期间最小化的函数。常见选择包括均方误差 (mse)、categorical_crossentropy 和 binary_crossentropy。损失函数由名称或通过从 tf.keras.losses 模块传递可调用对象来指定。
metrics:用于监控训练。它们是 tf.keras.metrics 模块中的字符串名称或可调用对象。

4使用.fit方法使模型与训练数据“拟合”;还有更多参数,例如:

model.fit(data, labels, epochs=10, batch_size=32,
          validation_data=(val_data, val_labels))

5.predict或evaluate方法进行预测。

详细信息可以在下面链接中找到:

https://tensorflow.google.cn/api_docs/python/tf/keras/Model#fit

(具体方法的使用有点复杂,我会在后面的博文中仔细分析学习)

 

 

  • 2
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值