給定權重w=[8, 10](1 x 2 向量)和b=-5一個超平面w^Tx b=0,我想使用這個超平面對測驗資料集X_te(51 x 2)進行分類。對于資料點x_i,分類基于 if w^Tx_i b >= 1then 分配標簽y=1和 if w^Tx_i b <= -1then 分配標簽y=-1。
我嘗試以下代碼:
def cla(X,w,b):
pre_te=np.zeros(len(X))
length=X.shape[0]
for i in range(length):
y_pred=np.dot(np.array(X[i]),w.T) b
if y_pred>=1:
pre_te[i]=1
else:
pre_te[i]=-1
return pre_te
我可以分配 1給y_predif y_pred>=1。如何添加-1到y_pred,如果y_pred<=-1在上面的for回圈?
我不認為寫else: pre_te[i]=-1作業......輸出中會有0......
uj5u.com熱心網友回復:
我認為你需要elif
for i in range(length):
y_pred=np.dot(np.array(X[i]),w.T) b
if y_pred>=1:
pre_te[i]=1
elif y_pred<=-1:
pre_te[i]=-1
return pre_te
轉載請註明出處,本文鏈接:https://www.uj5u.com/net/348110.html
上一篇:使用for回圈進行串列捆綁
下一篇:可能是一個關于c 回圈的簡單問題
