import numpy as np
from data_processor import DataProcessor
from neural_network import NeuralNetwork
if __name__ == '__main__':
data_processor = DataProcessor()
x_train, y_train = data_processor.get_train_set()
x_test, y_test = data_processor.get_test_set()
input_nodes = 2
hidden_nodes = 3
output_nodes = 1
network = NeuralNetwork(input_nodes=input_nodes,
hidden_nodes=hidden_nodes,
output_nodes=output_nodes,
lr=0.01)
network.train(x_train, y_train)
score = network.evaluate(x_test, y_test)
print(score)
x = np.array([[1, 1],
[10, 10],
[100, 100],
[2000, 1000]], dtype=float)
y = network.predict(x)
print(x)
