更新 function.py

2024-11-02 15:15:20 +08:00
parent da2746d141
commit e790be953e
1 changed files with 2 additions and 0 deletions
@@ -117,8 +117,10 @@ def load_data(filepath):
                values = [float(v) for v in values]
                if values[22] == 2 or values[22] == 3:
                    labels.append(0)
+                    data.append(values[:2] + values[3:23])
                elif values[22] == 1:
                    labels.append(1)
+                    data.append(values[:2] + values[3:23])
                else:
                    continue