Create Dataset class

This commit is contained in:
George Lacey 2017-02-10 12:12:16 +00:00
parent 9311411e15
commit d61b090d92

16
Dataset.py Normal file
View File

@ -0,0 +1,16 @@
from Data import Data
class Dataset(object):
def __init__(self, p_input, p_target):
self.inputs = {}
for index, item in enumerate(p_input):
input1 = float(p_input[index].split(',')[0].strip())
input2 = float(p_input[index].split(',')[1].strip())
if p_target[index] == "1":
output = 1
else:
output = -1
self.inputs[index] = Data(input1, input2, output)