验证深度学习开发环境是否成功搭建

最新推荐文章于 2024-08-23 15:11:12 发布

ZHANGHUIHUIA

最新推荐文章于 2024-08-23 15:11:12 发布

阅读量727

点赞数 1

分类专栏：安装文章标签：开发环境

本文链接：https://blog.csdn.net/ZHANGHUIHUIA/article/details/87922020

版权

本文档通过dk文件夹的创建及km.py脚本的使用，验证了一个深度学习开发环境是否已成功搭建。

摘要由CSDN通过智能技术生成

dk是我建立的一个文件夹，用来存放数据

cd dk
ls
python km.py

在这里插入图片描述

km.py

from __future__ import print_function

# 去除警告信息
import os
os.environ['TF_CPP_MIN_LOG_LEVEL']='2'

import keras
from keras.datasets import mnist
from keras.models import Sequential
from keras.layers import Dense, Dropout
from keras.optimizers import RMSprop, SGD

# the data, split between train and test sets
(x_train, y_train), (x_test, y_test) = mnist.load_data()

# 由于mnist的输入数据维度是(num, 28, 28)，这里需要把后面的维度直接拼起来变成784维
x_train = x_train.reshape(60000, 784) 
x_test = x_test.reshape(10000, 784)
x_train = x_train.astype('float32')
x_test = x_test.astype('float32'