Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import numpy as np
- from tensorflow.keras.models import Sequential
- from tensorflow.keras.layers import Dense
- train_x=np.arange(-20,20,0.25)
- train_y=np.sqrt((2*train_x**2)+1)
- model = Sequential()
- model.add(Dense(8,input_dim=1,activation='relu')) # input hidden ke input
- model.add(Dense(4, activation = 'relu')) # dim layer
- model.add(Dense(1, activation = 'linear')) # ouput layer
- model.compile (loss = 'mean_squared_error', optimizer = 'adam')
- model.fit(train_x,train_y,batch_size=20, epochs=10000,verbose=2)
- print(model.summary())
- model.save("model.h5")
- # test kasus
- x=np.array([26])
- predict= model.predict(x)
- print("f(26) =", predict )
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement