From 64286d6aa65ed2071fcf529a13a795b601937b46 Mon Sep 17 00:00:00 2001
From: George Lacey <490796@hull.ac.uk>
Date: Fri, 10 Feb 2017 12:12:16 +0000
Subject: [PATCH] Create Dataset class

---
 Dataset.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)
 create mode 100644 Dataset.py

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)