Skip to content

fialhocoelho/track-particles

This branch is 13 commits behind stonescenter/track-particles:master.

Folders and files

NameName
Last commit message
Last commit date

Latest commit

3bb42c5 · Feb 17, 2020

History

24 Commits
Feb 17, 2020
Feb 6, 2020
Feb 17, 2020
Feb 6, 2020
Feb 14, 2020
Feb 17, 2020
Feb 17, 2020
Feb 17, 2020
Feb 17, 2020
Feb 17, 2020
Feb 17, 2020
Feb 7, 2020

Repository files navigation

Track Particle

This project is the first version of Track Particles problem and it is part of SPRACE sponsored by Serrapilheira.

Run

To run:

  1. Clone our repository
  2. Configure your conda envirotment
  3. Run python main.py --config config.json. It will create a first model with LSTM architecture. If you don't have acces to a gpu, you could use the test_1_lstm_paralel.ipynb file to run into Google Colab.

Accuracy of Algorithm

We are using regressions metrics for each hit. Our accuracy is showed for forecas of 5 hit.

[Output] ---Regression Scores--- 
	R_2 statistics        (R2)  = 0.97
	Root Mean Square Error(RMSE) = 0.17
	Mean Absolute Error   (MAE) = 0.085

RMSE:		[0.170] 
RMSE features: 	[0.15, 0.23, 0.09] 
R^2  features:	[0.97, 0.94, 0.99] 

Vizualization

Open the plot_prediction.ipynb file to see the 5 hit predicted.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • HTML 57.7%
  • Jupyter Notebook 40.2%
  • Python 2.1%