SASRec.pytorch

基于PyTorch的SASRec模型实现

SASRec PyTorch 推荐系统自注意力机制序列推荐 Github 开源项目

SASRec.pytorch项目提供了自注意力序列推荐模型的PyTorch实现。相比原始TensorFlow版本，该项目优化了训练和推理流程，修复了正位置嵌入等问题。代码包含模型训练、评估和推理示例，并在MovieLens-1M数据集上展示了NDCG@10和HR@10指标的性能。项目适用于需要在PyTorch环境中研究或应用SASRec模型的人员，为推荐系统领域提供了有价值的开源资源。

访问官网

Github

介绍相关项目

keras-attention - Keras 兼容的注意力层，支持 Luong 和 Bahdanau 评分函数

Attention LayerBahdanauGithubKerasLuongTensorflow开源项目

Keras Attention Layer 支持 Luong 和 Bahdanau 的评分函数，与 Tensorflow 2.8 至 2.14 兼容。该层易于安装和使用，可根据需求调整参数，广泛应用于提高深度学习模型精度。提供丰富的实例和详细文档，包括在 LSTM 网络中的注意力机制实现，以及 IMDB 数据集和加数任务的实验数据。

equiformer-pytorch - SE(3)/E(3)等变注意力网络的高效PyTorch实现

AIEquiformerGATv2GithubSE3 Transformers开源项目深度学习

Equiformer-pytorch是一个基于PyTorch的SE(3)/E(3)等变注意力网络实现。该项目采用MLP注意力机制和非线性消息传递，实现了最先进的性能。它支持可逆网络以提高内存效率，并集成了最新的球谐函数稀疏化技术，大幅提升计算效率。Equiformer-pytorch还提供边缘和邻接矩阵支持，适用于蛋白质折叠等各种3D原子图任务。

pytorch-sentiment-neuron - Pytorch版本的情感神经元实现情感分析与文本生成

Githubcudamlstm_ns.ptpython 3.5pytorchsentiment开源项目

项目pytorch-sentiment-neuron基于Pytorch，实现了利用情感神经元进行情感分析和文本生成。用户可以通过预设模型文件和简单的命令行操作生成文本并进行情感分析，lm.py文件还允许在新数据上重新训练模型。该项目依赖Pytorch、Cuda和Python 3.5，适用于自然语言处理和情感分析领域的研究人员和开发者。

CoCa-pytorch - CoCa模型的PyTorch开源实现

CoCaGithubPytorch实现transformer架构图像文本模型对比学习开源项目

CoCa-pytorch项目提供了CoCa(Contrastive Captioners)模型的PyTorch实现。该项目将对比学习融入传统的编码器/解码器transformer，优化了图像到文本的转换。项目采用PaLM的transformer架构，包含单模态、多模态transformers和交叉注意力模块。这一实现为研究和开发图像-文本基础模型提供了有力工具。

Pytorch-RNN-text-classification - RNN短文本分类模型支持多类别高效处理

GithubLSTMPyTorchRNN开源项目短文本分类词嵌入

Pytorch-RNN-text-classification是一个多类别短文本分类模型，基于RNN架构设计。该项目使用Pytorch实现，集成词嵌入、LSTM（或GRU）和全连接层。模型支持GloVe预训练词向量，采用交叉熵损失函数和Adam优化器。通过零填充和PackedSequence技术处理mini-batch，提高训练效率。项目包含数据预处理和训练脚本，方便研究人员快速应用于实际文本分类任务。

3D-ResNets-PyTorch - 用于动作识别的 3D ResNets

3D ResNetsAction RecognitionGithubPyTorchSpatiotemporal 3D CNNs开源项目预训练模型

该项目提供基于PyTorch的3D ResNet代码，适用于动作识别，支持Kinetics和Moments in Time等数据集。项目包含训练、微调和测试脚本，并提供预训练模型，支持最新的PyTorch版本和分布式训练。用户可使用详细脚本进行数据准备和模型评估，适合研究与应用。

RecSysDatasets - 推荐系统公开数据集汇总及处理工具

GithubRecBole开源项目推荐系统数据处理数据集模型评估

RecSysDatasets是一个汇总公开推荐系统数据集的开源项目。该项目收集了电商、广告、电影等多个领域的数据集,并提供将数据集转换为统一格式的工具。这有助于研究人员更便捷地获取和使用各类推荐系统数据集,为算法开发和评估提供支持。项目与RecBole推荐系统库集成,便于进行算法测试。

Deep-reinforcement-learning-with-pytorch - 深度强化学习PyTorch实现与代码示例

DQNDeep Reinforcement LearningGithubGymTD3pytorch开源项目

本项目提供经典和前沿的深度强化学习算法PyTorch实现，包括DQN、DDPG、PPO等。项目持续更新并维护，适用于Anaconda虚拟环境管理。详细的安装步骤和测试方法确保用户能顺利运行代码，文档中还提供了相关论文和代码链接，便于深入学习研究。

flash-attention-minimal - 简化Flash Attention的CUDA和PyTorch最小化实现

CUDAFlash AttentionGPU加速GithubPyTorch开源项目注意力机制

flash-attention-minimal是一个开源项目，使用CUDA和PyTorch对Flash Attention进行最小化实现。项目仅用约100行代码完成前向传播，遵循原始论文符号表示，为CUDA初学者提供简明教程。通过与手动注意力机制的性能对比，展示了显著的速度提升。尽管目前存在一些限制，如缺少反向传播和固定块大小，但该项目为理解Flash Attention核心概念提供了有价值的参考资料。

Surprise - 专为推荐系统设计的Python科学计算工具包

GithubPythonSurprise协同过滤开源项目推荐系统机器学习

Surprise是一个专门用于构建和分析基于显式评分数据的推荐系统的Python科学计算工具包。它简化了数据集处理，提供多种预测算法和相似度度量，支持新算法实现，并具备评估和比较算法性能的工具。Surprise适用于学术研究和商业应用，为推荐系统开发提供了全面的解决方案。

相关项目

项目侧边栏1

项目侧边栏2

推荐项目

Project Cover

豆包MarsCode

豆包 MarsCode 是一款革命性的编程助手，通过AI技术提供代码补全、单测生成、代码解释和智能问答等功能，支持100+编程语言，与主流编辑器无缝集成，显著提升开发效率和代码质量。

Project Cover

AI写歌

Suno AI是一个革命性的AI音乐创作平台，能在短短30秒内帮助用户创作出一首完整的歌曲。无论是寻找创作灵感还是需要快速制作音乐，Suno AI都是音乐爱好者和专业人士的理想选择。

Project Cover

有言AI

有言平台提供一站式AIGC视频创作解决方案，通过智能技术简化视频制作流程。无论是企业宣传还是个人分享，有言都能帮助用户快速、轻松地制作出专业级别的视频内容。

Project Cover

Kimi

Kimi AI助手提供多语言对话支持，能够阅读和理解用户上传的文件内容，解析网页信息，并结合搜索结果为用户提供详尽的答案。无论是日常咨询还是专业问题，Kimi都能以友好、专业的方式提供帮助。

Project Cover

阿里绘蛙

绘蛙是阿里巴巴集团推出的革命性AI电商营销平台。利用尖端人工智能技术，为商家提供一键生成商品图和营销文案的服务，显著提升内容创作效率和营销效果。适用于淘宝、天猫等电商平台，让商品第一时间被种草。

Project Cover

吐司

探索Tensor.Art平台的独特AI模型，免费访问各种图像生成与AI训练工具，从Stable Diffusion等基础模型开始，轻松实现创新图像生成。体验前沿的AI技术，推动个人和企业的创新发展。

Project Cover

SubCat字幕猫

SubCat字幕猫APP是一款创新的视频播放器，它将改变您观看视频的方式！SubCat结合了先进的人工智能技术，为您提供即时视频字幕翻译，无论是本地视频还是网络流媒体，让您轻松享受各种语言的内容。

Project Cover

美间AI

美间AI创意设计平台，利用前沿AI技术，为设计师和营销人员提供一站式设计解决方案。从智能海报到3D效果图，再到文案生成，美间让创意设计更简单、更高效。

Project Cover

稿定AI

稿定设计是一个多功能的在线设计和创意平台，提供广泛的设计工具和资源，以满足不同用户的需求。从专业的图形设计师到普通用户，无论是进行图片处理、智能抠图、H5页面制作还是视频剪辑，稿定设计都能提供简单、高效的解决方案。该平台以其用户友好的界面和强大的功能集合，帮助用户轻松实现创意设计。

使用协议隐私政策广告服务

投诉举报邮箱: service@vectorlightyear.com

@2024 懂AI·鲁ICP备2024100362号-6·鲁公网安备37021002001498号