БИБЛИОТЕКА SURPRISE PYTHON

Библиотека surprise - это Python-библиотека, которая предоставляет удобные инструменты для создания, обучения и проверки рекомендательных систем. Это открытый и свободно распространяемый пакет, который может быть использован в коммерческих и некоммерческих целях.

Одним из главных достоинств библиотеки является ее простота использования. Это связано с тем, что она предлагает простой и интуитивно понятный API. Это упрощает жизнь разработчика и позволяет ему сосредоточиться на других аспектах создания рекомендательных систем.

Одним из примеров использования библиотеки surprise может быть создание рекомендательной системы, которая будет рекомендовать фильмы пользователям на основе их предпочтений. Ниже приведен пример кода на Python, который может выполнять эту задачу:

    from surprise import Dataset
    from surprise import Reader
    from surprise import KNNBasic
    
    # Определение базы данных
    reader = Reader(line_format='user item rating', sep=',', rating_scale=(1, 5))
    data = Dataset.load_from_file('ratings.csv', reader=reader)
    
    # Определение алгоритма
    sim_options = {'name': 'cosine', 'user_based': True}
    algo = KNNBasic(sim_options=sim_options)
    
    # Обучение модели
    trainset = data.build_full_trainset()
    algo.fit(trainset)
    
    # Получение рекомендаций для пользователя
    user_id = str(196)
    items_to_recc = algo.get_neighbors(trainset.to_inner_uid(user_id), k=3)
    
    # Вывод рекомендованных фильмов
    for item in items_to_recc:
        print(trainset.to_raw_iid(item))