def fitness(self, params=[0.1, 100, 10, 1, 0.8, 0.8, 0.1]): X = X_train y = y_train # 解压参数 learning_rate, n_estimators, max_depth, min_child_weight, subsample, colsample_bytree, gamma = params # 初始化模型 model = xgb.XGBRegressor( learning_rate=learning_rate, n_estimators=int(n_estimators), max_depth=int(max_depth), min_child_weight=int(min_child_weight), subsample=subsample, colsample_bytree=colsample_bytree, gamma=gamma, random_state=42, n_jobs=self.n_jobs ) model.fit(X, y) predictval=model.predict(X) print("R2 = ",metrics.r2_score(y_test,predictval)) # R2 return metrics.r2_score(y_test,predictval)

这段代码定义了一个计算适应度的函数fitness，其中传入一个参数params，包含了XGBoost模型的相关参数。在函数中，首先将训练数据X和目标数据y分别赋值为X_train和y_train，然后解压参数params，将其用于初始化一个XGBoost模型。接着，使用训练数据X和目标数据y来训练模型，并使用训练数据来进行预测，并计算预测结果与测试数据y_test之间的R2值。最后，将R2值作为适应度返回。