文件名称:tab-transformer-pytorch:在 Pytorch 中实现 TabTransformer,表格数据的注意力网络
文件大小:69KB
文件格式:ZIP
更新时间:2024-06-18 23:30:28
deep-learning tabular-data artificial-intelligence transformer attention-mechanism
标签转换器 在 Pytorch 中实现 ,表格数据的注意力网络。 这种简单的架构与 GBDT 的性能相差无几。 安装 $ pip install tab-transformer-pytorch 用法 import torch from tab_transformer_pytorch import TabTransformer cont_mean_std = torch . randn ( 10 , 2 ) model = TabTransformer ( categories = ( 10 , 5 , 6 , 5 , 8 ), # tuple containing the number of unique values within each category num_continuous = 10 , # number of co
【文件预览】:
tab-transformer-pytorch-main
----setup.py(767B)
----.gitignore(2KB)
----tab.png(64KB)
----tab_transformer_pytorch()
--------__init__.py(75B)
--------tab_transformer_pytorch.py(6KB)
----LICENSE(1KB)
----.github()
--------workflows()
----README.md(2KB)