簡體   English   中英

SVM 自定義 RBF 內核 IndexError

[英]SVM custom RBF kernel IndexError

我想為我的非線性可分數據實現 SVM RBF 內核。 我已經將我的內核定義為:

def rbf(va, vb):
    gamma = 0.7
    return exp(-gamma * linalg.norm(va - vb) ** 2)
clf = svm.SVC(kernel=rbf)
clf.fit(va, vb)

它顯示錯誤:

如果 X.shape[0] != X.shape[1]: IndexError: tuple index out of range

如何解決?

您的vavb形狀不匹配。 它們需要具有相同的形狀。 rbf 內核的語法應該是以下形式:

def rbf(va,vb) ,其中va是樣本的特征,比如 X 和vb是樣本的特征,比如 X`。

va 和 vb 不是特征和標簽。

您可以在 Github 上查看此鏈接以獲取類似的實現。 您可以看到他將所有樣本對傳遞給 rbf 內核,而不是樣本的特征和標簽。 轉到行號。 95.

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM