facebookresearch / facebookresearch/aepsych
Implement baseline linear model
- Dominant language
- HTML
- Stars
- 75
- Forks
- 58
- PR merge metrics
- No merged PRs in 30d
Description
Many classical psychophysics models are simple linear models or linear models with a polynomial basis. We should include them as baselines in AEPsych. For example, psignifit contains many of them (https://psignifit.sourceforge.net/PSYCHOMETRICFUNCTIONS.html). In the language of the psignifit docs, our "core" is a GP model and our sigmoids are the various objectives here https://github.com/facebookresearch/aepsych/blob/main/aepsych/acquisition/objective.py.
To implement a basic linear model code, we will need a class that extends `AEPsychMixin` that uses a latent linear model with learnable parameters instead of a GP. At minimum, it needs to support a `fit` method, and a `posterior` method so that we can integrate it with acquisition. This issue will require a bit of research to understand where and how to implement everything -- we are happy to support.
Contributor guide
Research direction
Start by reading aepsych/acquisition/objective.py and tracing how the existing GP model integrates with acquisition. Then locate AEPsychMixin and the current model interfaces to determine how a latent linear model should expose fit and posterior. Done means a baseline linear or polynomial-basis model supports both methods and can be integrated with acquisition.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- machine-learning
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 35/100