灰狼算法优化随机森林代码

以下是使用灰狼算法优化随机森林的 Python 代码：

import numpy as np
from sklearn.ensemble import RandomForestClassifier
from wolf_optimization import WOA  # 导入灰狼算法实现代码

# 定义随机森林超参数优化的目标函数
def objective_function(x):
    # x 是一个长度为 3 的向量，分别表示树的数量、树的深度和特征选择的方法
    n_estimators = int(x[0])
    max_depth = int(x[1])
    max_features = ['auto', 'sqrt', 'log2'][int(x[2])]
    clf = RandomForestClassifier(n_estimators=n_estimators, max_depth=max_depth, max_features=max_features)
    clf.fit(X_train, y_train)
    score = clf.score(X_val, y_val)
    return -score  # 因为 WOA 是求最小值，所以将 score 取负

# 定义随机森林超参数优化的边界
bounds = np.array([[10, 100], [5, 20], [0, 2]])

# 使用灰狼算法优化随机森林超参数
woa = WOA(objective_function, bounds)
best_score, best_params = woa.optimize()

# 使用最优的超参数训练随机森林模型
best_n_estimators = int(best_params[0])
best_max_depth = int(best_params[1])
best_max_features = ['auto', 'sqrt', 'log2'][int(best_params[2])]
clf = RandomForestClassifier(n_estimators=best_n_estimators, max_depth=best_max_depth, max_features=best_max_features)
clf.fit(X_train, y_train)
test_score = clf.score(X_test, y_test)
print(f'Test score: {test_score:.4f}')

其中 `X_train`, `y_train`, `X_val`, `y_val`, `X_test`, `y_test` 分别表示训练集、验证集和测试集的特征和标签。`WOA` 是灰狼算法的实现代码，可以参考相关文献或搜索相关代码实现。