Python(Classification)

Keras搭建NN(classification)

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from sklearn import datasets
from sklearn import preprocessing
from keras.models import Sequential,Model
from sklearn.model_selection import train_test_split
from keras.layers import Dense,Activation
from keras.optimizers import SGD,RMSprop
from sklearn.metrics import accuracy_score
'''
Dense():设置全连接
Activation():设置激活函数
'''
import warnings
warnings.filterwarnings('ignore')

dataset = datasets.load_iris()
x,y = dataset.data,dataset.target

nx = preprocessing.StandardScaler().fit_transform(x)
x_train,x_test,y_train,y_test = train_test_split(nx,y,test_size=0.2,random_state=1)
Y_train = (np.arange(3) == y_train[:,None]