Fix calculate shape stats to use same coeff columns (#9)

allen-cell-animated · Feb 17, 2023 · b422215 · b422215
1 parent 2ea6742
commit b422215
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/abm_shape_collection/calculate_shape_stats.py b/src/abm_shape_collection/calculate_shape_stats.py
@@ -11,16 +11,17 @@ def calculate_shape_stats(
 ) -> pd.DataFrame:
     all_stats = []
 
-    data_transform = pca.transform(data.filter(like="shcoeffs").values)
-    ref_data_transform = pca.transform(ref_data.filter(like="shcoeffs").values)
+    columns = ref_data.filter(like="shcoeffs").columns
+    data_transform = pca.transform(data[columns].values)
+    ref_data_transform = pca.transform(ref_data[columns].values)
 
     for component in range(components):
         ks_stats = get_ks_statistic(data_transform[:, component], ref_data_transform[:, component])
         ks_stats.update({"FEATURE": f"PC_{component + 1}", "TICK": np.nan})
         all_stats.append(ks_stats)
 
     for tick, tick_data in data.groupby("TICK"):
-        tick_data_transform = pca.transform(tick_data.filter(like="shcoeffs").values)
+        tick_data_transform = pca.transform(tick_data[columns].values)
 
         for component in range(components):
             tick_ks_stats = get_ks_statistic(