module harmonypy_gpu

---

function get_usage

get_usage(s)

---

function to_csr_cuda

to_csr_cuda(x, dtype)

Move to GPU as a csr_matrix.

---

function to_csc_cuda

to_csc_cuda(x, dtype)

Move to GPU as a csc_matrix, speed up column slice.

---

function get_dummies

get_dummies(x)

Return a sparse dummy matrix.

---

function run_harmony

run_harmony(
    data_mat: 'ndarray',
    meta_data: 'DataFrame',
    vars_use,
    init_seeds=None,
    theta=None,
    lamb=None,
    sigma=0.1,
    nclust=None,
    tau=0,
    block_size=0.05,
    max_iter_harmony=10,
    max_iter_kmeans=20,
    epsilon_cluster=1e-05,
    epsilon_harmony=0.0001,
    plot_convergence=False,
    verbose=True,
    reference_values=None,
    cluster_prior=None,
    n_init=1,
    random_state=0,
    dtype=<class 'numpy.float32'>
)

Run Harmony.

---

function safe_entropy

safe_entropy(x: 'array')

---

function moe_correct_ridge

moe_correct_ridge(Z_orig, Z_cos, Z_corr, R, W, K, Phi_Rk, Phi_moe, lamb)

---

class Harmony

method __init__

__init__(
    Z,
    init_seeds,
    n_init,
    Phi,
    Phi_moe,
    Pr_b,
    sigma,
    theta,
    max_iter_harmony,
    max_iter_kmeans,
    epsilon_kmeans,
    epsilon_harmony,
    K,
    block_size,
    lamb,
    verbose,
    random_state,
    dtype
)

---

method allocate_buffers

allocate_buffers()

---

method check_convergence

check_convergence(i_type)

---

method cluster

cluster()

---

method compute_objective

compute_objective()

---

method harmonize

harmonize(iter_harmony=10, verbose=True)

---

method init_cluster

init_cluster()

---

method kmeans_multirestart

kmeans_multirestart()

---

method result

result()

---

method update_R

update_R()

---

This file was automatically generated via lazydocs.