skbase_custom_estimator_with_param_getter_setter.py

1from skbase.base import BaseObject
2
3class MyEstimator(BaseObject):
4    def __init__(self, param1=1, param2="something"):
5        self.param1 = param1
6        self.param2 = param2
7        super().__init__()
8
9# Instantiate the object
10estimator = MyEstimator(param1=42, param2="hello")
11
12# Get parameters
13print(f"Parameters: {estimator.get_params()}")
14
15# Set parameters
16estimator.set_params(param1=100)
17print(f"Updated Parameters: {estimator.get_params()}")