#定义回调函数,以便使用早停技术并保持最佳模型

#!/usr/bin/env python
# -*- coding:utf-8 -*- 
# Author: Jia ShiLin

import numpy as np
import pandas as pd
from sklearn.model_selection import train_test_split

from keras.models import Sequential
from keras.layers import Dense
from keras.callbacks import EarlyStopping,ModelCheckpoint
from keras.optimizers import Adam
from sklearn.preprocessing import StandardScaler


SEED = 2019

#DATA
data = pd.read_csv('winequality-red.csv',sep=';')
y =data['quality']
x = data.drop(['quality',axis=1])

#拆分数据
X_train,X_test,y_train,y_test = train_test_split(X,y,test_size=0.2,random_state=SEED)
#显示平均品质和第一行训练集
print('Average quality training set:{:.4f}'.format(y_train.mean()))
X_train.head()

#数据标准化,
scalar = StandardScaler().fit(X_train)
X_train = pd.DataFrame(scalar.transform(X_train))
X_test = pd.DataFrame(scalar.transform(X_test))

#对每个验证输入的训练数据预测其平均质量
print('MSE:',np.mean((y_test-