PridaLab / rippl-AI

A open toolbox of several machine learning approaches for sharp-wave ripple detection
26 stars 4 forks source link

Small changes/upgrades #1

Closed JulioEI closed 1 year ago

JulioEI commented 1 year ago

Add raise ValueError when the user inputs a non-valid architecture name on predict(). Previously it raised a non-clear error downstream (UnboundLocalError: local variable 'y-predict' referenced before assignment) because it never went into any of the if statements before trying to return the prediction.

Added predictions_index to global scope within plot_predictions(), otherwise it prompts an error when trying to return the predictions since they are not on the proper scope.

Added a .gitignore to ignore pycache/