rnaglib.splitters.ClusterSplitter

class rnaglib.splitters.ClusterSplitter(similarity_threshold=0.3, n_jobs=-1, seed=0, *args, **kwargs)[source]

Abstract class for splitting by clustering with a similarity function

__init__(similarity_threshold=0.3, n_jobs=-1, seed=0, *args, **kwargs)[source]

Methods

__init__([similarity_threshold, n_jobs, seed])

cluster_split(dataset, frac[, n])

Fast cluster-based splitting adapted from ProteinShake (https://github.com/BorgwardtLab/proteinshake_release/blob/main/structure_split.py).