import matplotlib.pyplot as plt
import matplotlib as mpl
import numpy as np
import sklearn
import pandas as pd
import os
import sys
import time
import tensorflow as tf
from tensorflow import keras
# 数据读取
fashion_mnist = keras.datasets.fashion_mnist
(x_train_all, y_train_all), (x_test, y_test) = fashion_mnist.load_data()
x_valid, x_train = x_train_all[:5000], x_train_all[5000:]
y_valid, y_train = y_train_all[:5000], y_train_all[5000:]
# 归一化 x = (x - u)/std
from sklearn.preprocessing import StandardScaler
scaler = StandardScaler()
# X_train: [None, 28, 28] -> [None, 784], StandardScaler只能norm二维的,需要专程二维后再复原
# fit_transform 会保存训练集的均值方差,用于其他的归一化
x_train_scaled = scaler.fit_transform(
x_train.astype(np.float32).reshape(-1, 1)).reshape(-1, 28, 28)
# 验证集和测试集则直接transform
x_valid_scaled = scaler.transform(
x_valid.astype(np.float32).reshape(-1, 1)).reshape(-1, 28, 28)
x_test_scaled = scaler.transform(
x_test.astype(np
Tensorflow(四)搭建分类与回归模型模型
最新推荐文章于 2023-12-26 08:16:23 发布