Chit-Chat Dialogue Modeling through Sequence2Sequence
user1 | user2 |
---|---|
Hello! How are you doing? | Doing well, thanks! |
I got promoted today | Congratulations! |
Not doing well today | I’m sorry, can I do something to help you? |
ludwig experiment \
--dataset chitchat.csv \
--config_file config.yaml
With config.yaml
:
input_features:
-
name: user1
type: text
level: word
encoder: rnn
cell_type: lstm
reduce_output: null
output_features:
-
name: user2
type: text
level: word
decoder: generator
cell_type: lstm
attention: bahdanau
loss:
type: sampled_softmax_cross_entropy
training:
batch_size: 96