更新 function.py

2024-11-21 17:29:21 +08:00
parent 6f108089df
commit c53d627250
1 changed files with 6 additions and 1 deletions
@@ -62,12 +62,17 @@ def load_data(filepath):
    return X, y
 if __name__ == "__main__":
    # 以下代码可用于测试，可自行随意修改
    # 加载训练集
    X_train, y_train = load_data("./data/horse-colic.data")
    if X_train is None or y_train is None:
        print("训练数据加载失败，程序终止。")
        exit(1)
    # 加载测试集
    X_test, y_test = load_data("./data/horse-colic.test")
    if X_test is None or y_test is None:
        print("测试数据加载失败，程序终止。")
        exit(1)
    # 训练支持向量机模型
    weights, bias = train_svm(X_train, y_train, learning_rate=0.01, iterations=1000)