Spaces:
Sleeping
Sleeping
File size: 5,624 Bytes
10c7c36 |
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 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 |
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
|