keras.iris(예측 )

import numpy as np
import keras
from keras.models import Sequential
from keras.layers import Dense

# iris에 적용(값 예측)

# data 준비
from sklearn.datasets import load_iris

iris=load_iris()
X=iris.data[:,:3]
y=iris.data[:,3]

 

model=Sequential()
model.add(Dense(1,input_shape=(3,),activation='linear'))
model.compile(loss='mse',optimizer='sgd')

model.fit(X,y,epochs=1000)

 

ws=model.get_weights()
ws

[array([[-0.21790431],
        [ 0.22590208],
        [ 0.5348903 ]], dtype=float32), array([-0.1555938], dtype=float32)]

 

plt.plot(model.history.history['loss'])