facebookresearch / EGG

EGG: Emergence of lanGuage in Games
MIT License
286 stars 99 forks source link

decouple training and language analysis #150

Closed mbaroni closed 3 years ago

mbaroni commented 3 years ago

Is your proposal related to a problem?

Right now, language analysis is implemented with callback functions that operate as part of the training process. It would be cleaner and more efficient to decouple it, performing language analysis on trained agents.

Describe the solution you'd like to have implemented

After training, it's possible to save Sender and Receiver and query them with new data for language analysis. The language analysis functions generically operate on data structures such as message and sender_input, independently of their source.

Additional context

Solving this would probably also solve issue #148 (test set): merge?

robertodessi commented 3 years ago

merging with #148