File size: 476 Bytes
3d9ba5b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
"""
Abstractive Summarization Models (Seq2Seq with Attention & Transformer Wrappers)
"""

from .attention import BahdanauAttention, LuongAttention
from .encoder import Seq2SeqEncoder
from .decoder import Seq2SeqDecoder
from .seq2seq_model import Seq2SeqSummarizer
from .transformer_wrap import TransformerSummarizer

__all__ = [
    "BahdanauAttention",
    "LuongAttention",
    "Seq2SeqEncoder",
    "Seq2SeqDecoder",
    "Seq2SeqSummarizer",
    "TransformerSummarizer",
]