Tensorflow2.0将数据集进行归一化处理,提高精确度
在之前的TF2.0对Fashion_Mnist数据集分类这篇文章中,数据集并没有进行归一化,模型训练结果并不是很好,这次在数据集进行训练之前添加了归一化过程,模型精度提高的非常明显,具体的程序见下:
#-*- coding:utf-8 _*-
import matplotlib as mpl
import matplotlib.pyplot as plt
import numpy as np
import sklearn
import pandas as pd
import os
import sys
import time
import tensorflow as tf
from tensorflow import keras
# 输出库的名字和版本
print(sys.version_info)
for module in tf, mpl, np, pd, sklearn, tf, keras:
print(module.__name__, module.__version__)
# 指定GPU
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = '1'
config = tf.compat.v1.ConfigProto()
config