test

sample/model.py

@@ -64,6 +64,7 @@ class HDRModel:
         features = self.__getFeatures(xTrain)
 
         # Reshape input for random forest
+        # TO DO: sklearn split data to train and test
         xForestTrain = features.reshape(-1, features.shape[-1])
         yForestTrain = yTrain.reshape(-1)