Estou aprendendo a usar o módulo cerebral de Gekko para aplicativos de aprendizado profundo.
Eu tenho criado uma rede neural para aprender a função numpy.cos () e depois produzir resultados semelhantes.
Fico bem quando os limites do meu treinamento são:
x = np.linspace(0,2*np.pi,100)
Mas o modelo desmorona quando tento estender os limites para:
x = np.linspace(0,3*np.pi,100)
O que preciso mudar na minha rede neural para aumentar a flexibilidade do meu modelo para que ele funcione para outros limites?
Este é o meu código:
from gekko import brain
import numpy as np
import matplotlib.pyplot as plt
#Set up neural network
b = brain.Brain()
b.input_layer(1)
b.layer(linear=2)
b.layer(tanh=2)
b.layer(linear=2)
b.output_layer(1)
#Train neural network
x = np.linspace(0,2*np.pi,100)
y = np.cos(x)
b.learn(x,y)
#Calculate using trained nueral network
xp = np.linspace(-2*np.pi,4*np.pi,100)
yp = b.think(xp)
#Plot results
plt.figure()
plt.plot(x,y,'bo')
plt.plot(xp,yp[0],'r-')
plt.show()
Estes são os resultados para 2pi:
Estes são os resultados para 3pi: