update params to allow choosing best model for FT

Author: VipulMascarenhas

ads/aqua/finetuning/entities.py

@@ -25,6 +25,8 @@ class AquaFineTuningParams(DataClassSerializable):
     lora_target_modules: Optional[List] = None
     early_stopping_patience: Optional[int] = None
     early_stopping_threshold: Optional[float] = None
+    load_best_model_at_end: Optional[bool] = None
+    metric_for_best_model: Optional[bool] = None
 
 
 @dataclass(repr=False)

tests/unitary/with_extras/aqua/test_finetuning.py

@@ -302,6 +302,14 @@ def test_get_finetuning_default_params(self):
                 },
                 False,
             ),
+            (
+                {
+                    "epochs": 2,
+                    "load_best_model_at_end": True,
+                    "metric_for_best_model": "accuracy",
+                },
+                True,
+            ),
         ]
     )
     def test_validate_finetuning_params(self, params, is_valid):