一般来说,使用隐藏层神经元数量逐层减少,通常在步骤中每层除以2,
#!/usr/bin/env python
# -*- coding:utf-8 -*-
# Author: Jia ShiLin
'''
一般来说,使用隐藏层神经元数量逐层减少,通常在步骤中每层除以2,
'''
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import StandardScaler
from keras.models import Sequential
from keras.layers import Dense
from keras.optimizers import SGD
SEED = 7
# data
data = pd.read_csv('winequality-red.csv', sep=';')
x = data.drop(['quallity'], axis=1)
y = data['quality']
# 数据集划分
x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.2, random_state=SEED)
# 标准化
scaler = StandardScaler().fit(x_train)
x_train = pd.DataFrame(scaler.transform(x_train))
x_test = pd.DataFrame(s