diff options
Diffstat (limited to 'research/sander/experiment.go')
| -rw-r--r-- | research/sander/experiment.go | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/research/sander/experiment.go b/research/sander/experiment.go index 1521cf8..bcad99c 100644 --- a/research/sander/experiment.go +++ b/research/sander/experiment.go @@ -14,13 +14,15 @@ type Experiment struct { name string model string reference map[int64]struct{} + hiddenDim int } -func NewExperiment(name, model string, reference map[int64]struct{}) (*Experiment, error) { +func NewExperiment(name, model string, reference map[int64]struct{}, hiddenDim int) (*Experiment, error) { e := &Experiment{ name: name, model: model, reference: reference, + hiddenDim: hiddenDim, } return e, nil |