gpt4 book ai didi

Python for 循环查找 SVC 的最佳值(C 和 gamma)

转载 作者:行者123 更新时间:2023-11-30 09:33:53 25 4
gpt4 key购买 nike

我有一个数据集 X 和标签 y,用于训练 sklearn.SVC 模型并对其进行评分。数据分为 X_trainX_test。我运行 for 循环 来查找两个 SVC 参数的最佳可能值组合(即最佳分数):Cgamma。我可以打印出最高分数,但如何打印用于该特定分数的 C 和 Gamma 值?

for C in np.arange(0.05, 2.05, 0.05):
for gamma in np.arange(0.001, 0.101, 0.001):
model = SVC(kernel='rbf', gamma=gamma, C=C)
model.fit(X_train, y_train)
score = model.score(X_test, y_test)
if score > best_score:
best_score = score
print('Highest Accuracy Score: ', best_score)

最佳答案

您可以将其更改为:

   if score > best_score:
best_score = score
best_C = C
best_gamma = gamma

或者创建一个元组:

if score > best_score:
best_score = score, C, gamma

关于Python for 循环查找 SVC 的最佳值(C 和 gamma),我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/49006391/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com