Skip to content

Instantly share code, notes, and snippets.

@datalove
Last active August 29, 2015 14:23
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save datalove/2deaa8411262103d835b to your computer and use it in GitHub Desktop.
Save datalove/2deaa8411262103d835b to your computer and use it in GitHub Desktop.
Basic usage of 'caret' package with artificial neural net from 'nnet' package
#########################
# find and load packages
#########################
install.packages(c("nnet","caret"))
library(caret)
###################
# Build neural net
###################
formula <- mpg ~ cyl + disp + hp + wt + gear
net <- train(formula, data = mtcars, method = "nnet", linout = TRUE)
##################
# Use neural net
##################
new_data <- data.frame(cyl = 4, disp = 320, hp = 140, wt = 3.4, gear = 4)
predict(net, new_data) # predicts 25.67
######################
# Build linear regr,
# and predict
######################
lm1 <- lm(formula, data = mtcars)
predict(lm1, new_data) # predicts 23.49
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment