Requires Python >= 3.10:
conda create -y -n trust_learning python=3.10
conda activate trust_learning
Execute setup from the source code root of the repository:
cd src
pip install -e .
Figure 1: Visualization of the Proposed Framework
A markov decision process (MDP) environment script is located in the following directory:
cd src/environment
Human collection dataset of demonstrations is shared in the repository:
cd dataset/human_demonstrations
Training and testing datasets of .json files are located in the following directory:
cd dataset/human_demonstrations/2024_01_23_Train/jsons
cd dataset/human_demonstrations/2024_02_02_Test/jsons
Figure 3: Environment of Human-Robot Teaming in Construction (@ UTS - Robotics Institute) Figure 4: Human Demonstration Data Collection Process
Priorly collected dataset is shared in the repository:
cd dataset/human_demonstrations
The dataset is analyzed in the following notebook:
cd src/analyses
jupyter notebook
visualize_demonstration.ipynb
Neural network model files are located in the following directory:
cd src/models
To run the training script:
cd src/optimization
python train.py
To run the training of the policy model (explicitly):
cd src/optimization
python train_policy.py
To run the training of the reward model (explicitly):
cd src/optimization
python train_reward.py
The evaluation of the trained models is included in the following notebook:
cd src/evaluation
jupyter notebook
evaluate_trust_estimation.ipynb
A Beta Reputation System implementation script is located in the following directory:
cd src/trusts
model_dynamics.py
Figure 10: Illustration of an Iterative Trust Estimation Process
Visualize the results of the Modeled Human Trust in the following notebook:
cd src/evaluation
jupyter notebook
visualize_trust_dynamics.ipynb
The resultant Excel files of the experiments are shared in the following directory:
cd results/experiments
Trust learning stage experiment results:
cd results/experiments/learning_experiments
results/experiments/learning_stage_experiment_results.xlsx
Inference stage experiment results:
cd results/experiments/inference_experiments
results/experiments/inference_stage_experiment_results.xlsx
The trained policy network and reward model parameters are shared in the following directories:
cd results/policy_network_params
cd results/reward_network_params
For any inquiries or lack of clarity, please contact the corresponding author: ➔ [email protected]