双十二网站搜索 推荐

import pandas as pd
from sklearn.feature_extraction.text import TfidfVectorizer
from sklearn.metrics.pairwise import linear_kernel

# 假设有一个商品数据集
data = {
    'product_id': [1, 2, 3, 4],
    'name': ['Laptop', 'Smartphone', 'Tablet', 'Headphones'],
    'description': [
        'High performance laptop with 16GB RAM',
        'Latest smartphone with advanced camera features',
        'Portable tablet with long battery life',
        'Noise-cancelling headphones for immersive sound'
    ]
}

df = pd.DataFrame(data)

# 使用TF-IDF向量化商品描述
tfidf = TfidfVectorizer(stop_words='english')
df['description'] = df['description'].fillna('')
tfidf_matrix = tfidf.fit_transform(df['description'])

# 计算商品之间的相似度
cosine_sim = linear_kernel(tfidf_matrix, tfidf_matrix)

def get_recommendations(product_id, cosine_sim=cosine_sim):
    idx = df[df['product_id'] == product_id].index[0]
    sim_scores = list(enumerate(cosine_sim[idx]))
    sim_scores = sorted(sim_scores, key=lambda x: x[1], reverse=True)
    sim_scores = sim_scores[1:3]  # 获取最相似的两个商品
    product_indices = [i[0] for i in sim_scores]
    return df['name'].iloc[product_indices]

# 示例：获取与产品ID为1的商品相似的推荐商品
print(get_recommendations(1))