Fix runtime bug in Radius of Gyration of Doryab Location.

2020-10-10 13:26:06 -04:00 · 2020-10-10 13:26:06 -04:00 · dcf1255b4b
parent b0cbc0a54a
commit dcf1255b4b
1 changed files with 3 additions and 1 deletions
--- a/src/features/location_doryab/location_base.py
+++ b/src/features/location_doryab/location_base.py
@ -343,8 +343,10 @@ def radius_of_gyration(locationData,sampling_frequency):
        
        time_in_cluster = locationData[locationData["location_label"]==labels].shape[0]* sampling_frequency
        rog = rog + (time_in_cluster * distance)
-        
+    
    time_all_clusters = valid_clusters.shape[0] * sampling_frequency
+    if time_all_clusters == 0:
+        return 0
    final_rog = (1/time_all_clusters) * rog

    return np.sqrt(final_rog)