r-examples/market_basket_analysis_example.R

In Reza1317/funcly: Useful Functions in R for All-Purpose Analysis

#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#    https://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# author: Reza Hosseini

codePath = ""

source(paste0(codePath, "data_analysis.R"))
source(paste0(codePath, "clustering.R"))
source(paste0(codePath, "plotting.R"))
source(paste0(codePath, "market_basket_analysis.R"))

library("arules")
library("data.table")

n = 100
df = data.frame(
    "id"=rep("", n),
    "date"=rep("", n),
    "full_seq_basket"=rep("", n),
    stringsAsFactors=FALSE)

baskets = c("c;d;e", "e;f;g", "f;d;h", "a;b;c;d")
dates = c("2019-02-23", "2019-02-24", "2019-02-25")

for (i in 1:n) {
    df[i, ] =  list(i,  sample(dates, 1),  sample(baskets, 1))
}


res = Find_sigBaskets_fromSeqData(
    seqDf=df,
    idCols=c("id", "date"),
    basketCol="full_seq_basket",
    liftThresh=1.5,
    suppThresh=0.00001,
    confThresh=0.5,
    lhsSizeThresh=2,
    basketSizeLowerBound=1,
    keepFreqItemOnly=FALSE,
    topItemNum=200,
    writePath=NULL)