Skip to content

Commit

Permalink
Introducing Response Time Modeling (#361)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: #361

Implements https://arxiv.org/abs/2306.06296 (Response Time Improves Choice Prediction and Function Estimation for Gaussian Process Models of Perception and Preferences, UAI 2024).

Code to make figures etc in the paper will be available at https://github.com/facebookresearch/response-time-gps.

Differential Revision: D59561138
  • Loading branch information
mshvartsman committed Jul 15, 2024
1 parent 258a21f commit bcf5d13
Show file tree
Hide file tree
Showing 8 changed files with 1,289 additions and 0 deletions.
6 changes: 6 additions & 0 deletions aepsych/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@

from . import acquisition, config, factory, generators, models, strategy, utils
from .config import Config
from .distributions import RTDistWithUniformLapseRate, LogNormalDDMDistribution, ShiftedGammaDDMDistribution, ShiftedInverseGammaDDMDistribution, ShiftedLogNormalDDMDistribution
from .likelihoods import BernoulliObjectiveLikelihood
from .models import GPClassificationModel
from .strategy import SequentialStrategy, Strategy
Expand All @@ -31,6 +32,11 @@
"BernoulliObjectiveLikelihood",
"BernoulliLikelihood",
"GaussianLikelihood",
"RTDistWithUniformLapseRate",
"LogNormalDDMDistribution",
"ShiftedGammaDDMDistribution",
"ShiftedInverseGammaDDMDistribution",
"ShiftedLogNormalDDMDistribution",
]

try:
Expand Down
Loading

0 comments on commit bcf5d13

Please sign in to comment.