precola / tom

A Brain-Inspired Theory of Mind Spiking Neural Network for Reducing Safety Risks of Other Agents

ToM

This code accompanies the paper the paper "A Brain-Inspired Theory of Mind Spiking Neural Network for Reducing Safety Risks of Other Agents"

Requirments

• numpy
• scipy
• pytorch >= 1.7.0
• torchvision
• pygame

Run

Train

• the file to be run: main_both.py
• args:
  • the path to save net_NPC: --save_net_N
  • the path to save net_a: --save_net_a
  • time steps: --T

python main_both.py --save_net_N=net_NPC.pth --save_net_a=net_agent.pth --episodes=45 --trajectories=30 --T=50 --mode=train --task=both

Test

You can use the weigts saved by taining in the test environment.

python main_ToM.py --save_net_N=net_NPC.pth --save_net_a=net_agent.pth --episodes=45 --trajectories=30 --T=50 --mode=train --task=both

Citation

@article{zhao2022brain,
    title     = {A Brain-Inspired Theory of Mind Spiking Neural Network for Reducing Safety Risks of Other Agents},
    author    = {Zhao, Zhuoya and Lu, Enmeng and Zhao, Feifei and Zeng, Yi and Zhao, Yuxuan},
    journal   = {Frontiers in neuroscience},
    pages     = {446},
    year      = {2022},
    publisher = {Frontiers}
}