LabelMap
Generate label maps for ML/DL Projects
Usage
-
With CSV file
from labelmap import LabelMap label_map = LabelMap(labels_csv="path-to-csv", labels_col_name="labels") label_map.info() # print label map
-
With CSV file and exclude some labels
from labelmap import LabelMap label_map = LabelMap(labels_csv="path-to-csv", labels_col_name="labels", exclude_labels=["cat", "dog"]) label_map.info() # print label map
-
Without CSV file
from labelmap import LabelMap label_map = LabelMap() label_map.add("cat") label_map.add("dog") label_map.info() # print label map
LabelMap
class
Parameters in -
labels_csv
:str
- The path to the csv file containing the labels and ids.
- Not Required, default =
None
-
labels_col_name
:str
- The name of the column containing the labels.
- Required if
labels_col_name
is set, default =None
-
ids_col_name
:str
- The name of the column containing the ids.
- Not Required, default =
None
-
id_type
:int/float
- The type of the ids.
- Not Required, default =
int
-
exclude_labels
:list
- The labels to exclude from the label map.
- Not Required, default =
None
LabelMap
class
Methods in -
add(label)
- Add a label to the label map.
- Args:
label -> str
- Return Type:
None
-
force_get(label)
- Get the id of a label or if id not found then add the label and get.
- Args:
label -> str
- Return Type:
None
-
to_text(id)
- Get the label of an id.
- Args:
id -> int/ float
- Return Type:
str
-
to_id(label)
- Get the id of a label.
- Args:
label -> str
- Return Type:
int/ float
-
map()
- Get the label map as dict.
- Args:
None
- Return Type:
dict
-
labels()
- Get the list of labels.
- Args:
None
- Return Type:
list
-
info()
- Print the label map as table.
- Args:
None
- Return Type:
None
-
count()
- Get the number of labels.
- Args:
None
- Return Type:
int
-
as_dataframe()
- Label data dictionary to pandas dataframe.
- Args:
None
- Return Type:
pandas.DataFrame
-
save_csv(path)
- label data dictionary save to csv file.
- Args:
path -> str
- Return Type:
None