Commit 87c93ef2 authored by josh's avatar josh
Browse files

test

parent f8e02017
This diff is collapsed.
...@@ -64,6 +64,7 @@ class HDRModel: ...@@ -64,6 +64,7 @@ class HDRModel:
features = self.__getFeatures(xTrain) features = self.__getFeatures(xTrain)
# Reshape input for random forest # Reshape input for random forest
# TO DO: sklearn split data to train and test
xForestTrain = features.reshape(-1, features.shape[-1]) xForestTrain = features.reshape(-1, features.shape[-1])
yForestTrain = yTrain.reshape(-1) yForestTrain = yTrain.reshape(-1)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment