A language for synthesizing randomized experimental designs
SweetPea is a language for declaratively specifying randomized experimental designs, and a runtime for synthesizing trial sequences generated from the design specification; this prototype that is targeted at psychology and neuroscience experiments.
An experimental design is a description of experimental factors, relationships between factors, sequential constraints, and how to map those factors onto a sequence of trials. The reliability and validity of experimental results heavily relies on rigorous experimental design.
SweetPea provides a high-level interface to declaratively describe an experimental design, and a low-level synthesizer to generate unbiased sequences of trials given satisfiable constraints. SweetPea samples sequences of trials by compiling experimental designs into Boolean logic, which are then passed to a SAT-sampler. The SAT-sampler Unigen provides statistical guarantees that the solutions it finds are approximately uniformly probable in the space of all valid solutions. This means that while producing sequences of trials that are perfectly unbiased is intractable, we do the next best thing-- produce sequences that are approximately unbiased.
This project is at an early stage, and likely to change: it isn't yet ready for real-world useage. Please don't rely on any of this code!
SweetPea also depends on Docker being installed and running on your machine.
pip install sweetpea
import operator as op from sweetpea import * color = Factor("color", ["red", "blue"]) text = Factor("text", ["red", "blue"]) conLevel = DerivedLevel("con", WithinTrial(op.eq, [color, text])) incLevel = DerivedLevel("inc", WithinTrial(op.ne, [color, text])) conFactor = Factor("congruent?", [conLevel, incLevel]) design = [color, text, conFactor] crossing = [color, text] k = 1 constraints = [NoMoreThanKInARow(k, ("congruent?", "con"))] block = fully_cross_block(design, crossing, constraints) experiments = synthesize_trials(block) print_experiments(block, experiments)
Release history Release notifications
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
|Filename, size & hash SHA256 hash help||File type||Python version||Upload date|
|sweetpea-0.0.19-py3-none-any.whl (49.8 kB) Copy SHA256 hash SHA256||Wheel||py3|
|sweetpea-0.0.19.tar.gz (31.3 kB) Copy SHA256 hash SHA256||Source||None|