感知器的scala实现

object Perceptron{
val sourceX = Array(Array(3,3),Array(4,3),Array(1,1))
val resulty = Array(1,1,-1)
def perceptronSGD(X:Array[Array[Int]],y:Array[Int],learnRate:Double = 0.001,iterNum:Int = 1000):(Array[Double],Double)={
var theta:Array[Double] = new Array(X(0).length)
var b = 0.0
for(i<- 0 to iterNum){
val row = i % X.length
var rowSum = 0.0
for(col <- 0 until X(0).length){
rowSum += X(row)(col) * theta(col)
}
rowSum = (rowSum + b) * y(row)
if(!(rowSum > 0)){
for(col <- 0 until X(0).length){
theta(col) = theta(col) + learnRate * y(row) * X(row)(col)
}
b = b + learnRate * y(row)
}
}
(theta,b)
}
}