diff --git a/function.py b/function.py index b532259..4602e0a 100644 --- a/function.py +++ b/function.py @@ -89,7 +89,7 @@ def load_data(filepath): X = data[:, :-1] y = data[:, -1] - # 将标签转换为0和1(假设原始标签为1和2) + # 将标签转换为0和1)假设原始标签为1和2) y = np.where(y == 1, 0, 1) return X, y