随笔分类 -  推荐系统

摘要:一、main.py import torch import tqdm from sklearn.metrics import roc_auc_score from torch.utils.data import DataLoader import os import numpy as np from 阅读全文
posted @ 2021-08-17 14:02 jasonzhangxianrong 阅读(993) 评论(0) 推荐(0) 编辑
摘要:一、概述 https://blog.csdn.net/springtostring/article/details/108157070 FM模型善于挖掘二阶特征交叉关系,而神经网络DNN的优点是能够挖掘高阶的特征交叉关系,于是DeepFM将两者组合到一起,实验证明DeepFM比单模型FM、DNN效果 阅读全文
posted @ 2021-05-18 10:28 jasonzhangxianrong 阅读(1331) 评论(0) 推荐(0) 编辑
摘要:一、main函数 import torch import tqdm from sklearn.metrics import roc_auc_score from torch.utils.data import DataLoader from torchfm.dataset.avazu import 阅读全文
posted @ 2021-04-22 15:18 jasonzhangxianrong 阅读(490) 评论(0) 推荐(0) 编辑