Create Dataset class
This commit is contained in:
parent
9311411e15
commit
d61b090d92
16
Dataset.py
Normal file
16
Dataset.py
Normal 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)
|
Loading…
Reference in New Issue
Block a user