har_trees: Fix labels being cut off

Bad rstrip

Author: jonnor

examples/har_trees/har_data2labelstudio.py

@@ -34,7 +34,7 @@ def load_har_record(path,
             df['time'] = t
             df = df.set_index('time')
 
-            classname = f.split('_')[1].rstrip(suffix)
+            classname = os.path.splitext(f)[0].split('_')[1]
 
             # Remove :, special character on Windows
             filename = f.replace(':', '')

examples/har_trees/har_labelstudio2dataset.py

@@ -5,8 +5,8 @@
 import pandas
 
 
+from har_data2labelstudio import load_har_record
 
-from read_data import load_har_record
 
 def extract_filename(url):
 

examples/har_trees/har_quicklabel.py

@@ -45,15 +45,11 @@ def main():
     # Lookup data
     recordings = load_har_record(data_path)
 
-    print(recordings.columns)
-
-    print(recordings.head(5))
-
     # Create packed dataframe
     dfs = []
     for filename, row in recordings.iterrows():
         classname = row.classname
-        filename = filename.rstrip('.npy')
+        filename = filename
         #print(filename, classname)    
 
         d = row.data
@@ -66,7 +62,6 @@ def main():
     #p = 'data/processed/pamap2.parquet'
 
     out = pandas.concat(dfs, ignore_index=True)
-    print(out)
     out.to_parquet(out_path)
 
     #return
@@ -75,7 +70,7 @@ def main():
     print(df.columns)
     print(df.activity.value_counts())
     print(df.file.value_counts())
-    print(df.head())
+    print(df.head(5))
 
 
 if __name__ == '__main__':