更新 function.py

2024-11-02 15:48:50 +08:00
parent 275a84bc76
commit e2bbb3b69a
1 changed files with 4 additions and 1 deletions
@@ -90,7 +90,10 @@ def load_data(filepath):
    data = np.array(data)
    labels = np.array(labels)
    target_variable = np.where((labels == 2) | (labels == 3), 0, labels)
-    data = np.delete(data, [2] + list(range(23, 28)), axis=1)
+    
+    n_features = data.shape[1]
+    columns_to_delete = [2] + list(range(23, min(28, n_features)))
+    data = np.delete(data, columns_to_delete, axis=1)
    
    scaler = StandardScaler()
    data = scaler.fit_transform(data)