
TabTransformer-Pytorch: 在PyTorch中实现表格数据的注意力机制网络
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
TabTransformer-Pytorch是一款基于PyTorch框架开发的代码库,专注于构建用于处理表格数据的注意力机制模型。该工具集成了先进的机器学习技术,使研究人员和开发者能够便捷地应用自注意力架构于结构化数据上,推动了诸如分类、回归等任务中的性能提升。
标签转换器在 PyTorch 中实现的表格数据注意力网络与 GBDT 的性能相近。安装方法为:`pip install tab-transformer-pytorch`。用法如下:
```python
import torch
from tab_transformer_pytorch import TabTransformer
cont_mean_std = torch.randn(10, 2)
model = TabTransformer(categories=(10, 5, 6, 5, 8), num_continuous=10)
```
其中,`categories` 是一个元组,包含每个类别中唯一值的数量;`num_continuous` 表示连续特征的数量。
全部评论 (0)
还没有任何评论哟~


