From 7b222eb452358b59b8d9b5eee7ca541f5f4118c6 Mon Sep 17 00:00:00 2001 From: MilagrosMarin Date: Thu, 11 Apr 2024 02:34:16 +0200 Subject: [PATCH] `FullFitTask` to `Lookup` and add `fullfit_id`attr --- element_moseq/moseq_train.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/element_moseq/moseq_train.py b/element_moseq/moseq_train.py index b647f66..4c908d2 100644 --- a/element_moseq/moseq_train.py +++ b/element_moseq/moseq_train.py @@ -578,12 +578,13 @@ def make(self, key): @schema -class FullFitTask(dj.Manual): +class FullFitTask(dj.Lookup): """Insert the parameters for the full (Keypoint-SLDS model) fitting. The full model will generally require a lower value of kappa to yield the same target syllable durations. Attributes: PCAFit (foreign key) : `PCAFit` Key. + fullfit_id (smallint unsigned) : Unique ID for each full fitting task. full_latent_dim (int) : Latent dimension to use for the model full fitting. full_kappa (int) : Kappa value to use for the model full fitting. full_num_iterations (int) : Number of Gibbs sampling iterations to run in the model full fitting. @@ -593,10 +594,11 @@ class FullFitTask(dj.Manual): definition = """ -> PCAFit # `PCAFit` Key + fullfit_id : smallint unsigned # Unique ID for each full fitting task + --- full_latent_dim : int # Latent dimension to use for the model full fitting full_kappa : int # Kappa value to use for the model full fitting full_num_iterations : int # Number of Gibbs sampling iterations to run in the model full fitting - --- model_name : varchar(100) # Name of the model to be loaded if `task_mode='load'` task_mode='load' :enum('load','trigger')# Trigger or load the task full_fit_desc='' : varchar(1000) # User-defined description of the model full fitting task