Tensorflow implementation of the Rescorla-Wagner model
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
datasets
.gitignore
README.md
get-ndl-data.sh
model.py
rw.py
util.py

README.md

Rescorla-Wagner model

This implements the Rescorla-Wagner model in TensorFlow.

Example usage

To fetch datasets included in the NDL run:

./get-ndl-data.sh

Next, the training can be started:

./rw.py --data_file datasets/ndl/serbian.txt \
  --alpha_beta 0.001 --n_steps 1e7 --add_bias