闲聊对话建模通过 Sequence2Sequence

user1 user2
你好!你怎么样? 挺好的,谢谢!
我今天升职了 恭喜!
我今天过得不太好 我很抱歉,我能帮你做点什么吗?
ludwig experiment \
  --dataset chitchat.csv \
  --config config.yaml

使用 config.yaml

input_features:
    -
        name: user1
        type: text
        encoder: 
            type: rnn
            cell_type: lstm
            reduce_output: null

output_features:
    -
        name: user2
        type: text
        decoder: 
            type: generator
            cell_type: lstm
            attention: bahdanau
        loss:
            type: softmax_cross_entropy

trainer:
    batch_size: 96