rnaglib.tasks.RNAClassificationTask

class rnaglib.tasks.RNAClassificationTask(**kwargs)[source]
__init__(**kwargs)[source]

Methods

__init__(**kwargs)

compute_metrics(all_preds, all_probs, all_labels)

compute_one_metric(preds, probs, labels)

describe([recompute])

Get description of task dataset, including dimensions needed for model initialization and other relevant statistics.

dummy_inference()

evaluate(model, loader)

rtype:

dict

get_split_datasets([recompute])

get_split_loaders([recompute])

get_task_vars()

Define a FeaturesComputer object to set which input and output variables will be used in the task.

init_metadata()

Optionally adds some key/value pairs to self.metadata.

load()

Load dataset and splits from disk.

process()

Tasks must implement this method.

set_datasets([recompute])

Sets the train, val and test datasets Call this each time you modify self.dataset.

set_loaders([recompute])

Sets the dataloader properties.

split(dataset)

Calls the splitter and returns train, val, test splits.

write()

Save task data and splits to root.

Attributes

default_splitter

dummy_model

task_id

Task hash is a hash of all RNA ids and node IDs in the dataset