test/test_Softmax.R

In hetong007/rUFLDL: A pure R implementation of UFLDL

x = matrix(rnorm(10000),ncol=5)

y = x[,1]-x[,2]*x[,4]+x[,3]^2
y = findInterval(y,quantile(y,c(0.3,0.6)))

require(Matrix)
model = Softmax(x[1:1000,],y[1:1000],0.1,0.01,maxStep=500)

fitted = model[[3]]
pred = apply(model[[4]] %*% t(cbind(1,x[1001:2000,])),2,which.max)-1

table(as.factor(fitted),as.factor(y[1:1000]))
sum(diag(table(as.factor(fitted),as.factor(y[1:1000]))))/
    sum(table(as.factor(fitted),as.factor(y[1:1000])))
table(as.factor(pred),as.factor(y[1001:2000]))
sum(diag(table(as.factor(pred),as.factor(y[1001:2000]))))/
    sum(table(as.factor(pred),as.factor(y[1001:2000])))