增加trans prob输入

.vscode/launch.json

@@ -14,12 +14,11 @@
             "justMyCode": true,
             "args": ["--subj", "XW01", 
             "--n-trials", "15", 
-            // "--hand-feedback",
+            "--hand-feedback",
             "--com", "COM3", 
             "-fm", "flex", 
             "-vfr", "0.", 
-            "-scth", "0.75",
-            "--difficulty", "hard",
+            "--difficulty", "mid",
             "--model-path", "./static/models/XW01/riemann_rest+flex_12-05-2023-19-10-25.pkl"]
         },
         {
@@ -32,7 +31,8 @@
             "justMyCode": true,
             "args": ["--subj", "XW01", 
             "--com", "COM3", 
-            "-scth", "0.75",
+            "-scth", "0.9",
+            "-stp", "0.9",
             "--model-path", "./static/models/XW01/riemann_rest+flex_12-05-2023-19-10-25.pkl"]
         },
         {
@@ -67,7 +67,7 @@
             "cwd": "${workspaceFolder}/backend",
             "justMyCode": true,
             "args": ["--subj", "XW01", 
-            "--model-filename", "riemann_rest+flex_12-05-2023-19-10-25.pkl"]
+            "--model-filename", "riemann_rest+flex_12-06-2023-17-38-27.pkl"]
         },
         {
             "name": "Online simulation",
@@ -78,8 +78,9 @@
             "cwd": "${workspaceFolder}/backend",
             "justMyCode": true,
             "args": ["--subj", "XW01", 
-            "-scth", "0.75",
-            "--model-filename", "riemann_rest+flex_12-05-2023-19-10-25.pkl"]
+            "-scth", "0.9",
+            "-stp", "0.9",
+            "--model-filename", "riemann_rest+flex_12-06-2023-17-38-27.pkl"]
         },
         {
             "name": "Python: 当前文件",

backend/bci_core/online.py

@@ -22,15 +22,20 @@ class Controller:
     def __init__(self,
                  virtual_feedback_rate=1., 
                  model_path=None,
+                 state_trans_prob=0.8,
                  state_change_threshold=0.6):
         if (model_path is None) or (model_path == 'None'):
             self.real_feedback_model = None
         else:
             self.model_type, _ = parse_model_type(model_path)
             if self.model_type == 'baseline':
-                self.real_feedback_model = BaselineHMM(model_path, state_change_threshold=state_change_threshold)
+                self.real_feedback_model = BaselineHMM(model_path, 
+                state_trans_prob=state_trans_prob,
+                state_change_threshold=state_change_threshold)
             elif self.model_type == 'riemann':
-                self.real_feedback_model = RiemannHMM(model_path, state_change_threshold=state_change_threshold)
+                self.real_feedback_model = RiemannHMM(model_path, 
+                state_trans_prob=state_trans_prob,
+                state_change_threshold=state_change_threshold)
             else:
                 raise NotImplementedError
         self.virtual_feedback_rate = virtual_feedback_rate

backend/free_grasp.py

@@ -2,7 +2,7 @@
 # -*- coding: utf-8 -*-
 """
 This experiment was created using PsychoPy3 Experiment Builder (v2023.2.3),
-    on Wed Dec  6 17:54:49 2023
+    on Wed Dec  6 17:58:52 2023
 If you publish work using this script the most relevant publication is:
 
     Peirce J, Gray JR, Simpson S, MacAskill M, Höchenberger R, Sogo H, Kastman E, Lindeløv JK. (2019) 
@@ -76,6 +76,14 @@ def parse_args():
         type=float
     )
     parser.add_argument(
+        '--state-trans-prob',
+        '-stp',
+        dest='state_trans_prob',
+        help='Transition probability for HMM state change',
+        default=0.8,
+        type=float
+    )
+    parser.add_argument(
         '--model-path',
         dest='model_path',
         help='Path to model file',

backend/general_grasp_training.py

@@ -2,7 +2,7 @@
 # -*- coding: utf-8 -*-
 """
 This experiment was created using PsychoPy3 Experiment Builder (v2023.2.3),
-    on 十一月 29, 2023, at 12:36
+    on Wed Dec  6 17:50:50 2023
 If you publish work using this script the most relevant publication is:
 
     Peirce J, Gray JR, Simpson S, MacAskill M, Höchenberger R, Sogo H, Kastman E, Lindeløv JK. (2019) 
@@ -93,13 +93,6 @@ def parse_args():
         type=float
     )
     parser.add_argument(
-        '--state-change-threshold',
-        '-scth',
-        dest='state_change_threshold',
-        help='Threshold for HMM state change',
-        type=float
-    )
-    parser.add_argument(
         '--difficulty',
         help='Task difficultys',
         type=str
@@ -131,8 +124,7 @@ if args.hand_feedback:
 
 # build bci controller
 controller = Controller(args.virtual_feedback_rate, 
-                        args.model_path, 
-                        state_change_threshold=args.state_change_threshold)
+                        args.model_path)
 # Run 'Before Experiment' code from decision
 cnt_threshold_table = {
     'easy': 3,
@@ -217,7 +209,7 @@ def setupData(expInfo, dataDir=None):
     thisExp = data.ExperimentHandler(
         name=expName, version='',
         extraInfo=expInfo, runtimeInfo=None,
-        originPath='C:\\Users\\asena\\Desktop\\kraken\\backend\\general_grasp_training.py',
+        originPath='/Users/dingkunliu/Projects/MI-BCI-Proj/kraken/backend/general_grasp_training.py',
         savePickle=True, saveWideText=True,
         dataFileName=dataDir + os.sep + filename, sortColumns='time'
     )