Ejemplo n.º 1
0
from neuralNetwork import NeuralNetwork
from dataset import Dataset
import getArgs
import gc

args = getArgs.train()

batch_size = args["batch_size"]
dataset = Dataset(args["dataset"])
net = NeuralNetwork(args["epochs"], batch_size, args["learn_rate"], dataset)
net.train()
net.evaluate(dataset)
net.save_model(args["model"])
dataset.save_labels(args["le"])