添加EmotionRecognition模型注释
This commit is contained in:
parent
ee3dbb04f0
commit
25b99502fd
|
@ -9,6 +9,23 @@ model_args = {
|
||||||
}
|
}
|
||||||
|
|
||||||
class EmotionRecognition:
|
class EmotionRecognition:
|
||||||
|
"""
|
||||||
|
9-class emotion
|
||||||
|
0: angry
|
||||||
|
1: disgusted
|
||||||
|
2: fearful
|
||||||
|
3: happy
|
||||||
|
4: neutral
|
||||||
|
5: other
|
||||||
|
6: sad
|
||||||
|
7: surprised
|
||||||
|
8: unknown
|
||||||
|
return :
|
||||||
|
[{"emotion": "angry", "weight": },
|
||||||
|
{"emotion": "disgusted", "weight": },
|
||||||
|
...
|
||||||
|
]
|
||||||
|
"""
|
||||||
def __init__(self) -> None:
|
def __init__(self) -> None:
|
||||||
self.initialize(model_args=model_args)
|
self.initialize(model_args=model_args)
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue