network.py 文件源码

def train(self, x, y, learning_rate=0.01, epochs=200):
        optimizer = optimizers.adam(lr=learning_rate, decay=1e-6)
        self._model.compile(loss="mean_squared_error", optimizer=optimizer)

        self._model.fit(x, y, batch_size=32, validation_split=0.05, epochs=epochs, verbose=1)