Navigation Menu

Skip to content

Instantly share code, notes, and snippets.

@decisionmechanics
decisionmechanics / recommendation_engine_example.R
Last active June 28, 2018 12:15
Create item-based recommendations using a co-occurrence matrix
get_recommendation_ratings <- function(rating_file_path) {
# Read user ID, item ID, user preference CSV data
ratings <- read.csv(file = rating_file_path, header = FALSE, col.names = c('user', 'item', 'preference'))
# Create item co-occurrence matrix
co_occurrence_matrix <- crossprod(table(ratings[, c('user', 'item')]))
# Convert long format to wide format and replace NAs with 0s
user_ratings <- tidyr::spread(ratings, user, preference, fill = 0)
@decisionmechanics
decisionmechanics / spark_random_forest.R
Created March 21, 2017 18:56
Predicting wine quality using a random forest classifier in SparkR
library(readr)
library(dplyr)
url <- "https://archive.ics.uci.edu/ml/machine-learning-databases/wine-quality/winequality-white.csv"
df <-
read_delim(url, delim = ";") %>%
dplyr::mutate(taste = as.factor(ifelse(quality < 6, "bad", ifelse(quality > 6, "good", "average")))) %>%
dplyr::select(-quality)