src/main/Python/DatasetGenerator.py

   1 from chainer.datasets import tuple_dataset
   2 from Data import Data
   3
   4 class DatasetGenerator:
   5
   6     def generateDataset(self,dataList):
   7         graphdata = [] #type: List[float]
   8         labeldata = [] #type: List[float]
   9
  10         data ="" #type: Data
  11         for data in dataList:
  12             labeldata.append(data.getLable())
  13             graphdata.append(self.createGraphData(data.getData()))
  14
  15         return tuple_dataset(graphdata,labeldata)
  16
  17     def createGraphData(self,targetData):
  18
  19         result=[] #type: List[float]
  20
  21         for strline in targetData:
  22             data=strline.sprit;
  23             floatLine=[] #type: List[float]
  24
  25             for block in data:
  26                 floatLine.append(float(block))
  27             #TODO floatLineの変換処理(データ正規化)を入れておく
  28             result.extend(floatLine)
  29
  30         return result