Source code for Gated Multimodal Units for Information Fusion.
Get the following files and uncompress it in the root folder of this project:
Run the make script:
python3 make_dataset.py gmu.json
Getting more movies
You can extend the dataset by adding more IMDb IDs to the
links.csv file and run
get_data.py script to crawl other movies.
Train and eval the model
Generate random configurations:
python3 generators/gmu.py gmu.json
Train the model and then report performance in test set (e.g. best conf for GMU model #23):
python3 run.py json/gmu_23.json