Python load_ml_u1 примеры использования

Пример #1

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))

Пример #2

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)