public-projects
/
InferringIntention
2
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
InferringIntention/watch_and_help
History
Lei Shi e33a28f702 update README 2024-03-25 00:49:41 +01:00
..
checkpoints/train_strategy_full first commit 2024-03-24 23:42:27 +01:00
dataset first commit 2024-03-24 23:42:27 +01:00
scripts first commit 2024-03-24 23:42:27 +01:00
stan first commit 2024-03-24 23:42:27 +01:00
watch_strategy_full first commit 2024-03-24 23:42:27 +01:00
README.md update README 2024-03-25 00:49:41 +01:00

README.md

Watch And Help Dataset

Codes to reproduce results on WAH dataset1

Data

Extact dataset/watch_data.zip

Neural Network

Run sh scripts/train_watch_strategy_full.sh to train the model

To test model, either use trained model or extract checkpoints checkpoints/train_strategy_full/lstmlast.zip

Run sh scripts/test_watch_strategy_full.sh to test the model

Prediction Split

Create artificial users and sample predictions from 10% to 90%

cd stan
sh split_user.sh
sh sampler_user.sh

Bayesian Inference

Run inference to get results of user intention prediction and action length (0% to 100%) for all users

Rscript strategy_inference_test.R

Plot intention prediction results and 10% to 100% of actions results

sh plot_user_length.sh
sh plot_user_length_10_steps.sh

  1. Modified based on WAH train and test codes, (see WAH)[https://github.com/xavierpuigf/watch_and_help] ↩︎