https://github.com/thefloatingstring/nmt
https://github.com/thefloatingstring/nmt
Last synced: 7 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/thefloatingstring/nmt
- Owner: TheFloatingString
- Created: 2019-08-01T14:20:29.000Z (over 6 years ago)
- Default Branch: master
- Last Pushed: 2019-08-12T23:53:24.000Z (over 6 years ago)
- Last Synced: 2025-03-02T21:16:47.368Z (11 months ago)
- Language: Python
- Size: 3.38 MB
- Stars: 0
- Watchers: 2
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# NMT (NeuroTechX Machine Learning Tools)
### Installing requirements
`pip install requirements.txt`
### Running the code
BrainActivity takes in raw unprocessed data as input, that can either be of shape (4,), (1,4) or (x,4) where x is an int > 0.
BrainActivity then does an average of the input arrays and runs a normalization (using Sklearn).
Next, BrainActivity will run either a RandomForest prediction or a KNN prediction.
The function will return either `"left"`, `"right"` or `"rest"`
```python
from NMT.BrainActivity import *
X = [0,0,0,0] # dummy data
rf_predict(X) # returns either "left", "right", or "rest" using random forest classifier
knn_predict(X) # returns either "left", "right", or "rest" using knn classifier
```