| 86 | |
| 87 | @dataclass |
| 88 | class BoAmpsAlgorithm: |
| 89 | training_type: Optional[str] = None |
| 90 | algorithm_type: Optional[str] = None |
| 91 | algorithm_name: Optional[str] = None |
| 92 | algorithm_uri: Optional[str] = None |
| 93 | foundation_model_name: Optional[str] = None |
| 94 | foundation_model_uri: Optional[str] = None |
| 95 | parameters_number: Optional[float] = None |
| 96 | framework: Optional[str] = None |
| 97 | framework_version: Optional[str] = None |
| 98 | class_path: Optional[str] = None |
| 99 | layers_number: Optional[float] = None |
| 100 | epochs_number: Optional[float] = None |
| 101 | optimizer: Optional[str] = None |
| 102 | quantization: Optional[str] = None |
| 103 | |
| 104 | def to_dict(self) -> dict: |
| 105 | return _to_dict(self) |
| 106 | |
| 107 | @classmethod |
| 108 | def from_dict(cls, d: dict) -> "BoAmpsAlgorithm": |
| 109 | return cls(**{camel_to_snake(k): v for k, v in d.items()}) |
| 110 | |
| 111 | |
| 112 | @dataclass |
no outgoing calls
searching dependent graphs…