def setCFtype(self, cftype):
self.CFtype = cftype
foldername, filename = "Models", self.CFtype + "_CFModel.pkl"
if (self.fitModeAtInitialize == True):
if (AlgoBase.checkPathExists(self, foldername, filename) == True):
self.preds_df = AlgoBase.loadModel(self, foldername, filename)
else:
self.fit(similarity="cosine", CFtype="user")
self.fit(similarity="cosine", CFtype="item")
def __init__(self, dsReader, fitModeAtInitialize=True):
AlgoBase.__init__(self, dsReader, fitModeAtInitialize)
self.RECOMMENDER_NAME = "SVD Recommender"
foldername, filename = "Models", "SVDModel.pkl"
if (self.fitModeAtInitialize == True):
if (AlgoBase.checkPathExists(self, foldername, filename) == True):
self.preds_df = AlgoBase.loadModel(self, foldername, filename)
else:
self.fit()
def __init__(self, dsReader, fitModeAtInitialize=True):
AlgoBase.__init__(self, dsReader, fitModeAtInitialize)
self.RECOMMENDER_NAME = "TF=IDF Recommender"
foldername, filename = "Models", "TfidfModel.pkl"
if(self.fitModeAtInitialize == True):
if(AlgoBase.checkPathExists(self, foldername, filename) == True):
self.similarityM = AlgoBase.loadModel(self, foldername, filename)
else:
self.fit()