kaiwu.sampler package#

Module contents#

模块: sampler

功能: 提供一系列数据后处理工具

class kaiwu.sampler.SimulatedAnnealingSampler(initial_temperature=100, alpha=0.99, cutoff_temperature=0.001, iterations_per_t=10, size_limit=100, flag_evolution_history=False, verbose=False, rand_seed=None, process_num=1)#

基类：SimulatedAnnealingSamplerBase

求解Ising模型的模拟退火采样器。每个解都独立求解

参数:

  • initial_temperature (float) -- 初始温度.

  • alpha (float) -- 降温系数.

  • cutoff_temperature (float) -- 截止温度.

  • iterations_per_t (int) -- 每个温度迭代深度.

  • size_limit (int) -- 输出解的个数，默认输出100个解

  • flag_evolution_history (bool) -- 是否输出哈密顿量演化历史，默认False，当值为True时，通过get_ha_history方法获取演化历史

  • verbose (bool) -- 是否在控制台输出计算进度，默认False

  • rand_seed (int, optional) -- numpy随机数生成器的随机种子

  • process_num (int, optional) -- 并行进程数 (-1为自动调用所有可用核心，1为单进程). Defaults to 1.

示例

>>> import numpy as np
>>> import kaiwu as kw
>>> matrix = -np.array([[ 0. ,  1. ,  0. ,  1. ,  1. ],
...                     [ 1. ,  0. ,  0. ,  1.,   1. ],
...                     [ 0. ,  0. ,  0. ,  1.,   1. ],
...                     [ 1. ,  1.,   1. ,  0. ,  1. ],
...                     [ 1. ,  1.,   1. ,  1. ,  0. ]])
>>> worker = kw.classical.SimulatedAnnealingOptimizer(initial_temperature=100,
...                                                   alpha=0.99,
...                                                   cutoff_temperature=0.001,
...                                                   iterations_per_t=10,
...                                                   size_limit=10)
>>> # This value is random and cannot be predicted
>>> worker.solve(matrix) 
array([[-1,  1, -1,  1,  1],
       [-1,  1,  1, -1,  1],
       [-1,  1,  1, -1, -1],
       [ 1, -1, -1,  1,  1],
       [ 1, -1,  1, -1,  1],
       [ 1, -1,  1, -1, -1],
       [ 1, -1, -1, -1,  1],
       [ 1, -1,  1,  1, -1],
       [ 1,  1,  1, -1, -1],
       [-1, -1, -1,  1,  1]])
on_matrix_change()#

更新矩阵相关信息

get_ha_history()#

获取哈密顿量演化历史

返回:

哈密顿量随时间演化历史，key是时间，单位为秒，value是 hamilton

返回类型:

dict

single_process_solve(ising_matrix=None, init_solution=None, rand_seed=None, size_limit=None)#

单进程求解Ising矩阵

参数:

  • ising_matrix (np.ndarray, optional) -- Ising矩阵. Defaults to None.

  • init_solution (np.ndarray, optional) -- 初始解向量. Defaults to None.

  • rand_seed (int, optional) -- numpy随机数生成器的随机种子

返回:

解向量

返回类型:

np.ndarray

get_hamiltonian()#
返回:

当前解的哈密顿量值

返回类型:

hamiltonian (np.ndarray)

set_matrix(ising_matrix)#

设置矩阵并更新相关内容

solve(ising_matrix=None, negtail_flip=True, sort_solutions=False)#

求解Ising矩阵

参数:

  • ising_matrix (np.ndarray) -- Ising矩阵

  • negtail_flip (bool) -- 是否进行负尾翻转

  • sort_solutions (bool) -- 是否对解进行排序

返回:

解向量

返回类型:

output (np.ndarray)

solve_qubo(*args, **kwargs)#
On this page