Coverage for source/utils/optimizer_from_string_converter.py: 100%

1# utils/optimizer_from_string_converter.py 

2 

3from typing import Any, Type 

4from tensorflow.keras.optimizers import Adam, Optimizer 

5 

6from .base_from_string_converter import BaseFromStringConverter 

7 

8class OptimizerFromStringConverter(BaseFromStringConverter): 

9 """ 

10 Converts string identifiers to Keras optimizer classes. 

11 

12 This class implements a specific string-to-object conversion for Keras optimizer 

13 classes. It maps string names like 'adam' to their corresponding optimizer 

14 implementations. 

15 """ 

16 

17 def __init__(self, **kwargs: dict[str, Any]) -> None: 

18 """ 

19 Initializes the optimizer converter with options for optimizer configuration. 

20 

21 Parameters: 

22 **kwargs (dict[str, Any]): Optional parameters that will be passed to 

23 the constructor of the optimizer objects. 

24 """ 

25 

26 self._kwargs: dict[str, Any] = kwargs 

27 self._value_map: dict[str, Type[Optimizer]] = { 

28 'adam': Adam 

29 }