This is the third week of the project. Now that the main part of the project is complete, we have added the GRU method.
simple demostration of GRU algorithm
Now our project has two algorithms: LSTM and GRU. Both methods achieve acceptable prediction results.
In addition, we added a visualization of the predicted results to make it more intuitive to compare the predicted results with the actual results.
Finally, we adjusted the training parameters slightly based on the results, then we got the test prediction result as follows:
Prediction Result for LSTM:
Note that the prediction result may vary depends on training time and sets of data used.
The data set is divided into two parts. One part of the data will be used to train our algorithm (30%), and the other part of the function will be provided to the trained algorithm to predict battery life (70%).
评论
发表评论