evox.algorithms.so.pso_variants.fs_pso

模块内容

类

FSPSO

特征选择 PSO 算法

API

class evox.algorithms.so.pso_variants.fs_pso.FSPSO(pop_size: int, lb: torch.Tensor, ub: torch.Tensor, inertia_weight: float = 0.6, cognitive_coefficient: float = 2.5, social_coefficient: float = 0.8, mean=None, stdev=None, mutate_rate: float = 0.01, device: torch.device | None = None)

基础: evox.core.Algorithm

特征选择 PSO 算法

初始化

使用给定参数初始化FSPSO算法。

参数:

• pop_size -- 种群大小。

• lb -- 粒子位置的下界。必须是一个一维张量。

• ub -- 粒子位置的上界。必须是一个一维张量。

• inertia_weight -- 惯性权重。默认值为0.6。

• cognitive_coefficient -- 认知权重。默认为 2.5。

• social_coefficient -- 社会权重。默认值为 0.8。

• mean -- 正态分布的均值。默认为 None。

• stdev -- 正态分布的标准差。默认为 None。

• mutate_rate -- 变异率。默认为 0.01。

• device -- 设备的使用者（设备用于存放张量的硬件）。默认为 None。

init_step()

step()

使用 FSPSO 执行常规优化步骤。