本頁提供使用 AI 執行時建立推薦系統的筆記本範例。 這些範例示範如何使用現代深度學習方法建立高效的推薦模型。
| Tutorial | 說明 |
|---|---|
| 雙塔推薦模型 | 學習如何將推薦資料轉換成 Mosaic Data Shard（MDS）格式，然後利用這些資料建立雙塔推薦模型。 |
雙塔推薦系統
這些筆記本示範如何將建議資料轉換為 Mosaic Data Shard （MDS） 格式，然後使用該資料來建立雙塔建議模型。 這種方法對於大規模推薦系統特別有效。
資料準備：將推薦模型資料集轉換為 MDS 格式
首先，將您的推薦資料集轉換為 MDS 格式，以實現高效的資料載入：
轉換資料
模型訓練：使用 PyTorch Lightning 的雙塔式推薦模型
利用已準備好的資料集與 PyTorch Lightning Trainer API 在多個 GPU 節點（A10 或 H100 GPU）上訓練雙塔推薦模型。