Skip to content

Commit

Permalink
Merge pull request #7 from ElenaRyumina/main
Browse files Browse the repository at this point in the history
  • Loading branch information
@DmitryRyumin
DmitryRyumin authored Dec 12, 2023
2 parents 30c04ab + 82d2c06 commit 6e24d9a
Show file tree
Hide file tree
Showing 4 changed files with 29 additions and 29 deletions.
14 changes: 7 additions & 7 deletions oceanai/modules/lab/audio.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -140,11 +140,11 @@ def __post_init__(self):
super().__post_init__() # Выполнение конструктора из суперкласса

# Нейросетевая модель **tf.keras.Model** для получения оценок по экспертным признакам
self._audio_model_hc: Optional[keras.engine.functional.Functional] = None
self._audio_model_hc: Optional[keras.models.Functional] = None
# Нейросетевая модель **tf.keras.Model** для получения оценок по нейросетевым признакам
self._audio_model_nn: Optional[keras.engine.functional.Functional] = None
self._audio_model_nn: Optional[keras.models.Functional] = None
# Нейросетевые модели **tf.keras.Model** для получения результатов оценки персональных качеств
self._audio_models_b5: Dict[str, Optional[keras.engine.functional.Functional]] = dict(
self._audio_models_b5: Dict[str, Optional[keras.models.Functional]] = dict(
zip(self._b5["en"], [None] * len(self._b5["en"]))
)

Expand Down Expand Up @@ -178,7 +178,7 @@ def __post_init__(self):
# ------------------------------------------------------------------------------------------------------------------

@property
def audio_model_hc_(self) -> Optional[keras.engine.functional.Functional]:
def audio_model_hc_(self) -> Optional[keras.models.Functional]:
"""Получение нейросетевой модели **tf.keras.Model** для получения оценок по экспертным признакам
Returns:
Expand Down Expand Up @@ -238,7 +238,7 @@ def audio_model_hc_(self) -> Optional[keras.engine.functional.Functional]:
return self._audio_model_hc

@property
def audio_model_nn_(self) -> Optional[keras.engine.functional.Functional]:
def audio_model_nn_(self) -> Optional[keras.models.Functional]:
"""Получение нейросетевой модели **tf.keras.Model** для получения оценок по нейросетевым признакам
Returns:
Expand Down Expand Up @@ -298,7 +298,7 @@ def audio_model_nn_(self) -> Optional[keras.engine.functional.Functional]:
return self._audio_model_nn

@property
def audio_models_b5_(self) -> Dict[str, Optional[keras.engine.functional.Functional]]:
def audio_models_b5_(self) -> Dict[str, Optional[keras.models.Functional]]:
"""Получение нейросетевых моделей **tf.keras.Model** для получения результатов оценки персональных качеств
Returns:
Expand Down Expand Up @@ -943,7 +943,7 @@ def __concat_pred(

def __load_audio_model_b5(
self, show_summary: bool = False, out: bool = True
) -> Optional[keras.engine.functional.Functional]:
) -> Optional[keras.models.Functional]:
"""Формирование нейросетевой архитектуры модели для получения результата оценки персонального качества
.. note::
Expand Down
12 changes: 6 additions & 6 deletions oceanai/modules/lab/prediction.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -127,11 +127,11 @@ def __post_init__(self):
super().__post_init__() # Выполнение конструктора из суперкласса

# Нейросетевые модели **tf.keras.Model** для получения результатов оценки персональных качеств
self._av_models_b5: Dict[str, Optional[keras.engine.functional.Functional]] = dict(
self._av_models_b5: Dict[str, Optional[keras.models.Functional]] = dict(
zip(self._b5["en"], [None] * len(self._b5["en"]))
)

self._avt_model_b5: Optional[keras.engine.functional.Functional] = None
self._avt_model_b5: Optional[keras.models.Functional] = None

# ----------------------- Только для внутреннего использования внутри класса

Expand All @@ -150,7 +150,7 @@ def __post_init__(self):
# ------------------------------------------------------------------------------------------------------------------

@property
def av_models_b5_(self) -> Dict[str, Optional[keras.engine.functional.Functional]]:
def av_models_b5_(self) -> Dict[str, Optional[keras.models.Functional]]:
"""Получение нейросетевых моделей **tf.keras.Model** для получения результатов оценки персональных качеств
Returns:
Expand Down Expand Up @@ -223,7 +223,7 @@ def av_models_b5_(self) -> Dict[str, Optional[keras.engine.functional.Functional
return self._av_models_b5

@property
def avt_model_b5_(self) -> Optional[keras.engine.functional.Functional]:
def avt_model_b5_(self) -> Optional[keras.models.Functional]:
"""Получение нейросетевой модели **tf.keras.Model** для получения оценок персональных качеств
Returns:
Expand Down Expand Up @@ -567,7 +567,7 @@ def __load_model_weights(

def __load_avt_model_b5(
self, show_summary: bool = False, out: bool = True
) -> Optional[keras.engine.functional.Functional]:
) -> Optional[keras.models.Functional]:
"""Формирование нейросетевой архитектуры модели для получения оценок персональных качеств
.. note::
Expand Down Expand Up @@ -628,7 +628,7 @@ def __load_avt_model_b5(

def __load_av_model_b5(
self, show_summary: bool = False, out: bool = True
) -> Optional[keras.engine.functional.Functional]:
) -> Optional[keras.models.Functional]:
"""Формирование нейросетевой архитектуры модели для получения результата оценки персонального качества
.. note::
Expand Down
14 changes: 7 additions & 7 deletions oceanai/modules/lab/text.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -167,10 +167,10 @@ def __post_init__(self):
super().__post_init__() # Выполнение конструктора из суперкласса

# Нейросетевая модель **tf.keras.Model** для получения оценок по экспертным признакам
self._text_model_hc: Optional[keras.engine.functional.Functional] = None
self._text_model_hc: Optional[keras.models.Functional] = None
# Нейросетевая модель **tf.keras.Model** для получения оценок по нейросетевым признакам
self._text_model_nn: Optional[keras.engine.functional.Functional] = None
self._text_model_b5: Optional[keras.engine.functional.Functional] = None
self._text_model_nn: Optional[keras.models.Functional] = None
self._text_model_b5: Optional[keras.models.Functional] = None

# Словарь для формирования экспертных признаков
self._text_features: str = (
Expand Down Expand Up @@ -343,7 +343,7 @@ def __post_init__(self):
# ------------------------------------------------------------------------------------------------------------------

@property
def text_model_hc_(self) -> Optional[keras.engine.functional.Functional]:
def text_model_hc_(self) -> Optional[keras.models.Functional]:
"""Получение нейросетевой модели **tf.keras.Model** для получения оценок по экспертным признакам
Returns:
Expand All @@ -353,7 +353,7 @@ def text_model_hc_(self) -> Optional[keras.engine.functional.Functional]:
return self._text_model_hc

@property
def text_model_nn_(self) -> Optional[keras.engine.functional.Functional]:
def text_model_nn_(self) -> Optional[keras.models.Functional]:
"""Получение нейросетевой модели **tf.keras.Model** для получения оценок по нейросетевым признакам
Returns:
Expand All @@ -363,7 +363,7 @@ def text_model_nn_(self) -> Optional[keras.engine.functional.Functional]:
return self._text_model_nn

@property
def text_model_b5_(self) -> Optional[keras.engine.functional.Functional]:
def text_model_b5_(self) -> Optional[keras.models.Functional]:
"""Получение нейросетевой модели **tf.keras.Model** для получения оценок персональных качеств
Returns:
Expand Down Expand Up @@ -855,7 +855,7 @@ def __process_audio_and_extract_features(

def __load_text_model_b5(
self, show_summary: bool = False, out: bool = True
) -> Optional[keras.engine.functional.Functional]:
) -> Optional[keras.models.Functional]:
"""Формирование нейросетевой архитектуры модели для получения оценок персональных качеств
.. note::
Expand Down
18 changes: 9 additions & 9 deletions oceanai/modules/lab/video.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -150,13 +150,13 @@ def __post_init__(self):
super().__post_init__() # Выполнение конструктора из суперкласса

# Нейросетевая модель **tf.keras.Model** для получения оценок по экспертным признакам
self._video_model_hc: Optional[keras.engine.functional.Functional] = None
self._video_model_hc: Optional[keras.models.Functional] = None
# Нейросетевая модель **tf.keras.Model** для получения нейросетевых признаков
self._video_model_deep_fe: Optional[keras.engine.functional.Functional] = None
self._video_model_deep_fe: Optional[keras.models.Functional] = None
# Нейросетевая модель **tf.keras.Model** для получения оценок по нейросетевым признакам
self._video_model_nn: Optional[keras.engine.functional.Functional] = None
self._video_model_nn: Optional[keras.models.Functional] = None
# Нейросетевые модели **tf.keras.Model** для получения результатов оценки персональных качеств
self._video_models_b5: Dict[str, Optional[keras.engine.functional.Functional]] = dict(
self._video_models_b5: Dict[str, Optional[keras.models.Functional]] = dict(
zip(self._b5["en"], [None] * len(self._b5["en"]))
)

Expand Down Expand Up @@ -332,7 +332,7 @@ def __post_init__(self):
# ------------------------------------------------------------------------------------------------------------------

@property
def video_model_hc_(self) -> Optional[keras.engine.functional.Functional]:
def video_model_hc_(self) -> Optional[keras.models.Functional]:
"""Получение нейросетевой модели **tf.keras.Model** для получения оценок по экспертным признакам
Returns:
Expand Down Expand Up @@ -392,7 +392,7 @@ def video_model_hc_(self) -> Optional[keras.engine.functional.Functional]:
return self._video_model_hc

@property
def video_model_nn_(self) -> Optional[keras.engine.functional.Functional]:
def video_model_nn_(self) -> Optional[keras.models.Functional]:
"""Получение нейросетевой модели **tf.keras.Model** для получения оценок по нейросетевым признакам
Returns:
Expand Down Expand Up @@ -452,7 +452,7 @@ def video_model_nn_(self) -> Optional[keras.engine.functional.Functional]:
return self._video_model_nn

@property
def video_model_deep_fe_(self) -> Optional[keras.engine.functional.Functional]:
def video_model_deep_fe_(self) -> Optional[keras.models.Functional]:
"""Получение нейросетевой модели **tf.keras.Model** для получения нейросетевых признаков
Returns:
Expand Down Expand Up @@ -512,7 +512,7 @@ def video_model_deep_fe_(self) -> Optional[keras.engine.functional.Functional]:
return self._video_model_deep_fe

@property
def video_models_b5_(self) -> Dict[str, Optional[keras.engine.functional.Functional]]:
def video_models_b5_(self) -> Dict[str, Optional[keras.models.Functional]]:
"""Получение нейросетевых моделей **tf.keras.Model** для получения результатов оценки персональных качеств
Returns:
Expand Down Expand Up @@ -1192,7 +1192,7 @@ def __concat_pred(self, pred_hc: np.ndarray, pred_nn: np.ndarray, out: bool = Tr

def __load_video_model_b5(
self, show_summary: bool = False, out: bool = True
) -> Optional[keras.engine.functional.Functional]:
) -> Optional[keras.models.Functional]:
"""Формирование нейросетевой архитектуры модели для получения результата оценки персонального качества
.. note::
Expand Down

0 comments on commit 6e24d9a

Please sign in to comment.