diff --git a/Dataset.py b/Dataset.py new file mode 100644 index 0000000..864ee5b --- /dev/null +++ b/Dataset.py @@ -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)