🚀 P5 (Beauty Base)
將推薦作為語言處理:統一預訓練、個性化提示與預測範式
本項目採用統一的預訓練、個性化提示和預測範式來處理推薦問題,涵蓋順序推薦、直接推薦和解釋生成等多種功能,使用亞馬遜美國評論數據集進行訓練。
✨ 主要特性
- 順序推薦:依據用戶的購買歷史記錄,預測下一個可能推薦給用戶的商品。
- 直接推薦:從給定的商品列表中挑選最適合的商品推薦給用戶。
- 解釋生成:基於特徵詞為特定商品生成解釋說明。
- 文本生成:具備文本到文本的生成能力。
📚 詳細文檔
數據集
本模型使用的訓練數據為amazon_us_reviews
。
評估指標
模型評估使用了以下指標:
- NDCG:歸一化折損累積增益
- HR:命中率
- MAE:平均絕對誤差
- BLUE:雙語評估替補
- ROUGE:自動文摘評估指標
示例
順序推薦
I find the purchase history list of user_823 :
5255 -> 3001 -> 3771 -> 2973
I wonder what is the next item to recommend to the user . Can you help me decide ?
直接推薦
Pick the most suitable item from the following list and recommend to user_182 :
5871 , 3575 , 6355 , 3665 , 7968 , 1054 , 11837 , 9031 , 2643 , 3125 , 11476 , 1529 , 6300 , 11755 , 9410 , 1578 , 5953 , 5042 , 10881 , 2221 , 11286 , 10458 , 2081 , 3722 , 10581 , 5879 , 1780 , 7411 , 5202 , 2082 , 82 , 8131 , 77 , 8097 , 1053 , 9946 , 1341 , 4508 , 2613 , 629 , 4869 , 9833 , 7076 , 6178 , 6679 , 6650 , 472 , 8821 , 4005 , 4184 , 2866 , 4988 , 10759 , 6358 , 4137 , 790 , 5390 , 9330 , 3691 , 2667 , 5620 , 11982 , 4799 , 10062 , 4278 , 4530 , 7944 , 10225 , 1766 , 6657 , 11371 , 305 , 1091 , 7144 , 1869 , 744 , 295 , 91 , 6947 , 9290 , 2977 , 11206 , 1677 , 7812 , 1159 , 1128 , 8762 , 5795 , 8061 , 9639 , 6161 , 2142 , 8124 , 5316 , 10425 , 12097 , 476 , 5710 , 1802 , 8969
解釋生成
Based on the feature word shampoo , generate an explanation for user_837 about this product : Dove Nourishing Oil Shampoo, 25.4 Ounce
模型圖示

📄 許可證
本項目採用MIT許可證。