Skip to content

subsetnames

subsetnames

Module for the subset names

Classes

SubsetNames

names for datasets in datasets.yml

Functions

get_eval_save_names 

get_eval_save_names()

Returns the names for the evaluation dataset names

Source code in niceml/config/subsetnames.py 
28
29
30
def get_eval_save_names() -> List[str]:
    """Returns the names for the evaluation dataset names"""
    return [get_save_name(x) for x in EVAL_DATASET_LIST]

get_save_name 

get_save_name(dataset_name)

Removes only the data_ prefix

Source code in niceml/config/subsetnames.py 
14
15
16
17
18
def get_save_name(dataset_name: str) -> str:
    """Removes only the data_ prefix"""
    if dataset_name.startswith("data_"):
        return dataset_name[5:]
    return dataset_name