cyto_dl.callbacks.tabular_saver module

class cyto_dl.callbacks.tabular_saver.SaveTabularData(save_dir, meta_keys=[], as_parquet: bool = True, save_suffix: str | None = None, col_prefix: str = 'feat')

Bases: Callback

Callback to save tabular data to disk as a .csv or .parquet after prediction.

Parameters:

• save_dir (str) – directory to save the tabular data

• meta_keys (list) – list of keys in the metadata to include as columns in the saved data

• as_parquet (bool) – Saves data as parquet if True, otherwise saves as csv

• save_suffix (str) – suffix to add to the saved file name

• col_prefix (str) – prefix to add to the column names of the saved data

on_predict_epoch_end(trainer, pl_module)

pred_to_df(pred)

save_feats(predictions, stage)