File size: 1,697 Bytes
de38b77
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
class Customization:
    def __init__(self):
        self.alert_thresholds = {}
        self.notification_methods = {}
        self.custom_alert_rules = []
        self.input_processing_settings = {}
        self.model_settings = {}

    def set_alert_threshold(self, alert_type, threshold):
        self.alert_thresholds[alert_type] = threshold

    def get_alert_threshold(self, alert_type):
        return self.alert_thresholds.get(alert_type, 0.8)

    def customize_notification_method(self, method, custom_function):
        self.notification_methods[method] = custom_function

    def get_notification_methods(self):
        return list(self.notification_methods.keys())

    def create_custom_alert_rule(self, rule):
        self.custom_alert_rules.append(rule)

    def get_custom_alert_rules(self):
        return self.custom_alert_rules

    def set_input_processing_setting(self, setting, value):
        self.input_processing_settings[setting] = value

    def get_input_processing_setting(self, setting):
        return self.input_processing_settings.get(setting, None)

    def enable_model(self, model_name):
        self.model_settings[model_name] = {"enabled": True}

    def disable_model(self, model_name):
        self.model_settings[model_name] = {"enabled": False}

    def set_model_setting(self, model_name, setting, value):
        if model_name not in self.model_settings:
            self.model_settings[model_name] = {}
        self.model_settings[model_name][setting] = value

    def get_model_setting(self, model_name, setting):
        if model_name in self.model_settings:
            return self.model_settings[model_name].get(setting, None)
        return None