Create Dataset class
This commit is contained in:
parent
878c2c857a
commit
64286d6aa6
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