update

neuromodulation · May 3, 2023 · 12ac1b6 · 12ac1b6
1 parent e2fe4cf
commit 12ac1b6
Showing 1 changed file with 108 additions and 0 deletions.
diff --git a/src/plot_utils.py b/src/plot_utils.py
@@ -928,3 +928,111 @@ def plot_rate_scores_on(rate, scores):
     plt.title('STN full beta ')
 
 
+def plot_duration_scores_on(duration, scores):
+
+    fig = plt.figure(35)
+    plt.scatter(duration, scores)
+    plt.xlabel('mean burst duration')
+    plt.ylabel('UPDRS Score ON')
+    plt.title('STN full beta ')
+    return fig
+
+def plot_amplitude_scores_off(amplitude, scores):
+
+    fig = plt.figure(36)
+    plt.scatter(amplitude, scores)
+    plt.xlabel('mean burst amplitude')
+    plt.ylabel('UPDRS Score OFF')
+    plt.title('STN full beta ')
+    return fig
+
+def plot_amplitude_scores_on(amplitude, scores):
+
+    fig = plt.figure(37)
+    plt.scatter(amplitude, scores)
+    plt.xlabel('mean burst amplitude')
+    plt.ylabel('UPDRS Score ON')
+    plt.title('STN full beta ')
+    return fig
+
+def plot_rate_scores_off(rate, scores):
+
+    fig = plt.figure(38)
+    plt.scatter(rate, scores)
+    plt.xlabel('mean burst rate')
+    plt.ylabel('UPDRS Score OFF')
+    plt.title('STN full beta ')
+    return fig
+
+def plot_rate_scores_on(rate, scores):
+
+    fig = plt.figure(39)
+    plt.scatter(rate, scores)
+    plt.xlabel('mean burst rate')
+    plt.ylabel('UPDRS Score ON')
+    plt.title('STN full beta ')
+    return fig
+
+def plot_duration_diff(diff, scores):
+
+    fig = plt.figure(40)
+    plt.scatter(diff, scores)
+    plt.xlabel('mean burst duration')
+    plt.ylabel('UPDRS Score diff')
+    plt.title('STN full beta ')
+    return fig
+
+def plot_amplitude_diff(diff, scores):
+
+    fig = plt.figure(41)
+    plt.scatter(diff, scores)
+    plt.xlabel('mean burst amplitude')
+    plt.ylabel('UPDRS Score diff')
+    plt.title('STN full beta ')
+    return fig
+
+def plot_rate_diff(diff, scores):
+
+    fig = plt.figure(42)
+    plt.scatter(diff, scores)
+    plt.xlabel('mean burst rate')
+    plt.ylabel('UPDRS Score diff')
+    plt.title('STN full beta ')
+    return fig
+
+def plot_prol_off(diff, scores):
+
+    fig = plt.figure(43)
+    plt.scatter(diff, scores)
+    plt.xlabel('prolonged bursts')
+    plt.ylabel('UPDRS Score OFF')
+    plt.title('STN full beta ')
+    return fig
+
+def plot_prol_on(diff, scores):
+
+    fig = plt.figure(44)
+    plt.scatter(diff, scores)
+    plt.xlabel('prolonged bursts')
+    plt.ylabel('UPDRS Score ON')
+    plt.title('STN full beta ')
+    return fig
+
+def plot_prol_diff(diff, scores):
+
+    fig = plt.figure(45)
+    plt.scatter(diff, scores)
+    plt.xlabel('prolonged bursts')
+    plt.ylabel('UPDRS Score diff')
+    plt.title('STN full beta ')
+    return fig
+
+def plot_dur_ampl(duration, amplitude):
+    fig = plt.figure(46)
+    plt.scatter(duration, amplitude)
+    plt.xlabel('duration')
+    plt.ylabel('Amplitude')
+    plt.title('STN full beta')
+    return fig
+
+