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         print(labeldata)
  16         return tuple_dataset(graphdata,labeldata)
  17
  18     def createGraphData(self,targetData):
  19
  20         result=[] #type: List[float]
  21
  22         for strline in targetData:
  23             print(strline)
  24
  25             floatLine=[] #type: List[float]
  26
  27             for block in strline:
  28                 floatLine.append(float(block))
  29             #TODO floatLineの変換処理(データ正規化)を入れておく
  30             result.extend(floatLine)
  31
  32         return result