add private module to create model object for movement api

This commit is contained in:
jordi.hasianta
2022-09-15 19:46:33 +07:00
parent ded93afb86
commit 760c9617be
6 changed files with 125 additions and 0 deletions

View File

@@ -0,0 +1,18 @@
from opendbm.api_lib.model import VideoModel
from opendbm.dbm_lib.dbm_features.raw_features.movement.voice_tremor import run_vtremor
class VocalTremor(VideoModel):
def __init__(self):
super().__init__()
self._params = [
"mov_freqtremfreq",
"mov_freqtremindex",
"mov_freqtrempindex",
"mov_amptremfreq",
"mov_amptremindex",
"mov_amptrempindex",
]
def _fit_transform(self, path):
return run_vtremor(path, ".", self.r_config, save=False)