rnaglib.tasks.RNAClassificationTask¶
- class rnaglib.tasks.RNAClassificationTask(**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