データセット生成機能を追加

[stux/ultron.git] / src / main / Python / DatasetGenerator.py

diff --git a/src/main/Python/DatasetGenerator.py b/src/main/Python/DatasetGenerator.py
index 47270ea..ca0a8af 100644 (file)
--- a/src/main/Python/DatasetGenerator.py
+++ b/src/main/Python/DatasetGenerator.py
@@ -1,2 +1,30 @@
 from chainer.datasets import tuple_dataset
+from Data import Data
 
+class DatsetGenerator:
+
+    def generateDataset(self,dataList):
+        graphdata = [] #type: List[float]
+        labeldata = [] #type: List[float]
+
+        data ="" #type: Data
+        for data in dataList:
+            labeldata.append(data.getLable())
+            graphdata.append(self.createGraphData(data.getData()))
+
+        return tuple_dataset(graphdata,labeldata)
+
+    def createGraphData(self,targetData):
+
+        result=[] #type: List[float]
+
+        for strline in targetData:
+            data=strline.sprit;
+            floatLine=[] #type: List[float]
+
+            for block in data:
+                floatLine.append(float(block))
+            #TODO floatLineの変換処理(データ正規化)を入れておく
+            result.extend(floatLine)
+
+        return result
\ No newline at end of file