main.py

from neuroevolution.networks.basic_neural_network import BasicNeuralNetwork
from neuroevolution.networks.genetic_neural_network import GeneticNeuralNetwork
from neuroevolution.networks.annealed_neural_network import AnnealedNeuralNetwork
from neuroevolution.networks.strategy_neural_network import StrategyNeuralNetwork
import numpy as np

if __name__ == "__main__":
    feature_set = np.array([[0,1],[0,0],[1,0],[1,1]])
    labels = np.array([[0,0,0,1]])
    labels = labels.reshape(4,1)
    # print(output_layer)
    #Creating the actual nnet
    nnet = StrategyNeuralNetwork(layers=[2,1], input_size=2, num_of_classes= 1)
    # print(nnet.params)
    # print(nnet.population.keys())
    # print(nnet.feed_forward)
    nnet.train(feature_set,labels, 200)
    vs = nnet.population.values()
    vs = list(vs)
    for i in vs: print(i['loss'])
    print(nnet.population)
    # print(nnet.population)
    # print(nnet.params)
    # print(nnet.loss)