adding sample std if non bernouilli rewards (#56)

* adding sample std if non bernouilli rewards

* switching to ddof=1

Author: TLSDC

src/agentlab/analyze/inspect_results.py

@@ -242,9 +242,23 @@ def get_std_err(df, metric):
     data = df[metric].dropna().values
 
     # asser either 0 or 1
-    assert np.all(np.isin(data, [0, 1]))
+    if np.all(np.isin(data, [0, 1])):
+        mean = np.mean(data)
+        std_err = np.sqrt(mean * (1 - mean) / len(data))
+    else:
+        return get_sample_std_err(df, metric)
+    return mean, std_err
+
+
+def get_sample_std_err(df, metric):
+    """Get the standard error for a binary metric."""
+    # extract non missing values
+    data = df[metric].dropna().values
+
     mean = np.mean(data)
-    std_err = np.sqrt(mean * (1 - mean) / len(data))
+    std_err = np.std(data, ddof=1) / np.sqrt(len(data))
+    if np.isnan(std_err):
+        std_err = 0
     return mean, std_err