skbase_baseobject_custom_class_with_params_and_tags.py

1from skbase.base import BaseObject
2
3class MyObject(BaseObject):
4    """A custom object inheriting from skbase BaseObject."""
5
6    def __init__(self, param1=1, param2="something"):
7        self.param1 = param1
8        self.param2 = param2
9        super().__init__()
10
11# Instantiate the object
12my_obj = MyObject(param1=42, param2="hello")
13
14# Get parameters (scikit-learn compatible get_params)
15params = my_obj.get_params()
16print(f"Parameters: {params}")
17
18# Set parameters (scikit-learn compatible set_params)
19my_obj.set_params(param1=100)
20print(f"Updated param1: {my_obj.param1}")
21
22# Objects can also have tags (metadata)
23my_obj._set_tags(my_tag="example_value")
24tag_value = my_obj.get_tag("my_tag")
25print(f"Tag value: {tag_value}")