Spaces:
Sleeping
Sleeping
class Container: | |
def __init__(self, x_train=None, y_train=None, x_test=None, y_test=None, hyper_params_optimize=None): | |
self.x_train = x_train | |
self.y_train = y_train | |
self.x_test = x_test | |
self.y_test = y_test | |
self.hyper_params_optimize = hyper_params_optimize | |
self.info = {"参数": {}, "指标": {}} | |
self.y_pred = None | |
self.train_sizes = None | |
self.train_scores_mean = None | |
self.train_scores_std = None | |
self.test_scores_mean = None | |
self.test_scores_std = None | |
self.status = None | |
self.model = None | |
def get_info(self): | |
return self.info | |
def set_info(self, info: dict): | |
self.info = info | |
def set_y_pred(self, y_pred): | |
self.y_pred = y_pred | |
def get_data_fit_values(self): | |
return [ | |
self.y_pred, | |
self.y_test | |
] | |
def get_learning_curve_values(self): | |
return [ | |
self.train_sizes, | |
self.train_scores_mean, | |
self.train_scores_std, | |
self.test_scores_mean, | |
self.test_scores_std | |
] | |
def set_learning_curve_values(self, train_sizes, train_scores_mean, train_scores_std, test_scores_mean, test_scores_std): | |
self.train_sizes = train_sizes | |
self.train_scores_mean = train_scores_mean | |
self.train_scores_std = train_scores_std | |
self.test_scores_mean = test_scores_mean | |
self.test_scores_std = test_scores_std | |
def get_status(self): | |
return self.status | |
def set_status(self, status: str): | |
self.status = status | |
def get_model(self): | |
return self.model | |
def set_model(self, model): | |
self.model = model | |
class PaintObject: | |
def __init__(self): | |
self.color_cur_num = 0 | |
self.color_cur_list = [] | |
self.label_cur_num = 0 | |
self.label_cur_list = [] | |
self.x_cur_label = "" | |
self.y_cur_label = "" | |
self.name = "" | |
def get_color_cur_num(self): | |
return self.color_cur_num | |
def set_color_cur_num(self, color_cur_num): | |
self.color_cur_num = color_cur_num | |
def get_color_cur_list(self): | |
return self.color_cur_list | |
def set_color_cur_list(self, color_cur_list): | |
self.color_cur_list = color_cur_list | |
def get_label_cur_num(self): | |
return self.label_cur_num | |
def set_label_cur_num(self, label_cur_num): | |
self.label_cur_num = label_cur_num | |
def get_label_cur_list(self): | |
return self.label_cur_list | |
def set_label_cur_list(self, label_cur_list): | |
self.label_cur_list = label_cur_list | |
def get_x_cur_label(self): | |
return self.x_cur_label | |
def set_x_cur_label(self, x_cur_label): | |
self.x_cur_label = x_cur_label | |
def get_y_cur_label(self): | |
return self.y_cur_label | |
def set_y_cur_label(self, y_cur_label): | |
self.y_cur_label = y_cur_label | |
def get_name(self): | |
return self.name | |
def set_name(self, name): | |
self.name = name | |
class SelectModel: | |
def __init__(self): | |
self.models = None | |
self.waterfall_number = None | |
self.force_number = None | |
self.beeswarm_plot_type = None | |
self.dependence_col = None | |
self.data_distribution_col = None | |
self.data_distribution_is_rotate = None | |
self.descriptive_indicators_col = None | |
self.descriptive_indicators_is_rotate = None | |
self.heatmap_col = None | |
self.heatmap_is_rotate = None | |
def get_heatmap_col(self): | |
return self.heatmap_col | |
def set_heatmap_col(self, heatmap_col): | |
self.heatmap_col = heatmap_col | |
def get_heatmap_is_rotate(self): | |
return self.heatmap_is_rotate | |
def set_heatmap_is_rotate(self, heatmap_is_rotate): | |
self.heatmap_is_rotate = heatmap_is_rotate | |
def get_models(self): | |
return self.models | |
def set_models(self, models): | |
self.models = models | |
def get_waterfall_number(self): | |
return self.waterfall_number | |
def set_waterfall_number(self, waterfall_number): | |
self.waterfall_number = waterfall_number | |
def get_force_number(self): | |
return self.force_number | |
def set_force_number(self, force_number): | |
self.force_number = force_number | |
def get_beeswarm_plot_type(self): | |
return self.beeswarm_plot_type | |
def set_beeswarm_plot_type(self, beeswarm_plot_type): | |
self.beeswarm_plot_type = beeswarm_plot_type | |
def get_dependence_col(self): | |
return self.dependence_col | |
def set_dependence_col(self, dependence_col): | |
self.dependence_col = dependence_col | |
def get_data_distribution_col(self): | |
return self.data_distribution_col | |
def set_data_distribution_col(self, data_distribution_col): | |
self.data_distribution_col = data_distribution_col | |
def get_data_distribution_is_rotate(self): | |
return self.data_distribution_is_rotate | |
def set_data_distribution_is_rotate(self, data_distribution_is_rotate): | |
self.data_distribution_is_rotate = data_distribution_is_rotate | |
def get_descriptive_indicators_is_rotate(self): | |
return self.descriptive_indicators_is_rotate | |
def set_descriptive_indicators_is_rotate(self, descriptive_indicators_is_rotate): | |
self.descriptive_indicators_is_rotate = descriptive_indicators_is_rotate | |
def get_descriptive_indicators_col(self): | |
return self.descriptive_indicators_col | |
def set_descriptive_indicators_col(self, descriptive_indicators_col): | |
self.descriptive_indicators_col = descriptive_indicators_col | |