def main():
a = DataManager()
a.loadData()
b = NeuralNetwork()
b.createModel()
b.train(a.train_data,
a.train_labels,
a.eval_data,
a.eval_labels,
epochs=100)
b.saveModel("test.h5")
def main():
a = DataManager()
a.loadData()
b = NeuralNetwork()
b.createModel()
b.train(a.train_data, a.train_labels, epochs=100)
res = b.evaluate(a.eval_data, a.eval_labels)
print("Accuracy : {}".format(res))
b.saveModel()
pass
def main():
data_manager = DataManager()
data_manager.loadData()
neural_network = NeuralNetwork()
neural_network.create_model()
neural_network.train(data_manager.train_data,
data_manager.train_labels,
data_manager.eval_data,
data_manager.eval_labels,
epochs=100)
result = neural_network.evaluate(data_manager.eval_data,
data_manager.eval_labels)
neural_network.save_model()
print("Accuracy : {}".format(result))
pass