|
|
@@ -13,7 +13,6 @@ from sklearn.metrics import accuracy_score
|
|
|
from dataloaders import neo
|
|
|
import bci_core.online as online
|
|
|
import bci_core.utils as bci_utils
|
|
|
-import bci_core.viz as bci_viz
|
|
|
from settings.config import settings
|
|
|
|
|
|
|
|
|
@@ -100,7 +99,6 @@ def _evaluation_loop(raw, events, model_hmm, step_length, event_trial_length):
|
|
|
stim_pred = _event_to_stim_channel(events_pred, len(raw.times))
|
|
|
stim_pred_naive = _event_to_stim_channel(events_pred_naive, len(raw.times))
|
|
|
|
|
|
- # TODO: auc
|
|
|
accu_hmm = accuracy_score(stim_true, stim_pred)
|
|
|
accu_naive = accuracy_score(stim_true, stim_pred_naive)
|
|
|
|
