🎨 improve structure

tongplw · Jul 9, 2020 · b04648b · b04648b
1 parent f6091f2
commit b04648b
Show file tree

Hide file tree

Showing 7 changed files with 5 additions and 6 deletions.
diff --git a/tester.py → dev.py b/tester.py → dev.py
@@ -4,13 +4,12 @@
 import matplotlib.pyplot as plt
 
 from tqdm import tqdm
-from muse import MuseMonitor
-
+from src.muse import MuseMonitor
 
 if __name__ == "__main__":
 
     headset = MuseMonitor(debug=True)
-    df = pd.read_csv('data.csv', header=None).iloc[:, 1:]
+    df = pd.read_csv('res/data.csv', header=None).iloc[:, 1:]
     df.columns = ['delta', 'theta', 'low-alpha', 'high-alpha', 'low-beta', 'high-beta', 'low-gamma', 'mid-gamma']
 
     atts = []
@@ -19,6 +18,6 @@
         waves = df.iloc[i].to_dict()
         attention = headset._attention(waves)
         atts += [attention]
-
+        
     plt.hist(atts, bins=200)
     plt.show()
diff --git a/graph.py b/graph.py
diff --git a/main.py b/main.py
@@ -1,6 +1,6 @@
 import time
 import pandas as pd
-from muse import MuseMonitor
+from src.muse import MuseMonitor
 from datetime import datetime
 
 

diff --git a/requirements.txt b/requirements.txt
diff --git a/data.csv → res/data.csv b/data.csv → res/data.csv
diff --git a/scaler → res/scaler b/scaler → res/scaler
diff --git a/muse.py → src/muse.py b/muse.py → src/muse.py
@@ -19,7 +19,7 @@ def __init__(self, server=None, port=None, debug=False):
         self._buffer = []
         self._attention_buff = [.5, .5, .5, .5, .5]
         self._attention_history = []
-        self.scaler = joblib.load('scaler')
+        self.scaler = joblib.load('res/scaler')
 
         self.raw = Value('d', 0)
         self.attention = Value('d', 0)