def main(): #load MovieLens u1 dataset #had to change directory, so we can run the called function in recsys/ rather than recsys/item_similarity/ os.chdir('..') ml_u1 = datasets.load_ml_u1() os.chdir('item_similarity') print(calculate_mae(ml_u1.test.predictions_df))
def main(): #load MovieLens dataset #had to change directory, so we can run the called function in recsys/ rather than recsys/item_similarity/ os.chdir('..') #ml_100k = datasets.load_ml_100k() ml_u1 = datasets.load_ml_u1() os.chdir('item_similarity') prediction = predict(ml_u1, 'filename') print(prediction)