Skip to content

Commit fc81adb

Browse files
author
Shimanto
authored
Add files via upload
1 parent d9ff715 commit fc81adb

1 file changed

Lines changed: 29 additions & 0 deletions

File tree

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,29 @@
1+
library(MASS)
2+
##install.packages('corrplot')
3+
library(corrplot)
4+
library(glmnet)
5+
data("nlschools")
6+
str(nlschools)
7+
nlschools$class<-NULL
8+
p.cor<-cor(nlschools[,-5])
9+
corrplot.mixed(p.cor)
10+
ind<-sample(2,nrow(nlschools),replace=T,prob = c(0.7,0.3))
11+
train<-nlschools[ind==1,]
12+
test<-nlschools[ind==2,]
13+
train$COMB<-model.matrix( ~ COMB - 1, data=train ) #convert to dummy variable
14+
test$COMB<-model.matrix( ~ COMB - 1, data=test )
15+
predictor_variables<-as.matrix(train[,2:4])
16+
language_score<-as.matrix(train$lang)
17+
lasso<-glmnet(predictor_variables,language_score,family="gaussian",alpha=1)
18+
print(lasso)
19+
plot(lasso,xvar = "lambda",label = T)
20+
21+
plot(lasso,xvar='dev',label=T)
22+
test.matrix<-as.matrix(test[,2:4])
23+
lasso.y<-predict(lasso,newx = test.matrix,type = 'response',s=.02)
24+
plot(lasso.y,test$lang)
25+
lasso.resid<-lasso.y-test$lang
26+
mean(lasso.resid^2)
27+
28+
29+

0 commit comments

Comments
 (0)