Commit 99f27d9b authored by Kevin Richard Doyle's avatar Kevin Richard Doyle
Browse files

Tried to fix nn.py to train network.

parent 4cad9f78
......@@ -2,8 +2,28 @@ from tensorflow import keras
import numpy as np
import pandas as pd
from tensorflow.keras import layers
from keras.models import Sequential
from keras.layers import Dense
class nn(object):
def __init__(self, sizes):
self.numlayers=len(sizes)
input_list = []
output_list = []
def __init__(self, size, optimizer, input_list, output_list):
self.numlayers=sizes
self.optimizer= optimizer
self.input_list= input_list
self.output_list = output_list
model= Sequential()
model.add(Dense(12,activation= 'relu'))
model.add(Dense(8,activation='relu'))
model.add(Dense(2, activation='relu'))
model.compile(loss='binary_crossentropy',optimizer='adam',)
model.fit(input_list, epochs=150, batch_size=10)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment