INNER CODE UNIT · Python

training

BigDataBiology/SemiBin · SemiBin/main.py:1100

def training(logger, contig_fasta,
             data, data_split, cannot_link,
             *, output, device, mode,
             args):
    """
    Training the model

    model: [single/several]
    """
    from .semi_supervised_model import train_semi
    from .self_supervised_model import train_self
    import pandas as pd

    if mode == 'single':
        logger.info('Start training from a single sample.')
        data_ = pd.read_csv(data[0], index_col=0)
        col_name = data_.columns[-1].split('_')[-1]
        is_combined = col_name != 'var'

View source record →