--- title: AutoInt keywords: fastai sidebar: home_sidebar summary: "A pytorch implementation of AutoInt." description: "A pytorch implementation of AutoInt." nb_path: "nbs/models/models.autoint.ipynb" ---
{% raw %}
{% endraw %} {% raw %}
{% endraw %} {% raw %}

class AutoInt[source]

AutoInt(field_dims, embed_dim, atten_embed_dim, num_heads, num_layers, mlp_dims, dropouts, has_residual=True) :: Module

A pytorch implementation of AutoInt. Reference: W Song, et al. AutoInt: Automatic Feature Interaction Learning via Self-Attentive Neural Networks, 2018.

{% endraw %} {% raw %}
{% endraw %}

References:-- W Song, et al. AutoInt: Automatic Feature Interaction Learning via Self-Attentive Neural Networks, 2018. https://arxiv.org/abs/1810.11921.