actableai.classification.models package

Submodules

actableai.classification.models.tabpfn module

class actableai.classification.models.tabpfn.TabPFNModel(**kwargs)

Bases: autogluon.core.models.abstract.abstract_model.AbstractModel

TabPFN classification Model

get_memory_size() → int

classmethod load(path: str, reset_paths=True, verbose=True)

Loads the model from disk to memory.

Parameters

• path (str) – Path to the saved model, minus the file name. This should generally be a directory path ending with a ‘/’ character (or appropriate path separator value depending on OS). The model file is typically located in path + cls.model_file_name.
• reset_paths (bool, default True) – Whether to reset the self.path value of the loaded model to be equal to path. It is highly recommended to keep this value as True unless accessing the original self.path value is important. If False, the actual valid path and self.path may differ, leading to strange behaviour and potential exceptions if the model needs to load any other files at a later time.
• verbose (bool, default True) – Whether to log the location of the loaded file.

Returns

model – Loaded model object.

Return type

cls

save(path: Optional[str] = None, verbose=True) → str

Saves the model to disk.

Parameters

• path (str, default None) – Path to the saved model, minus the file name. This should generally be a directory path ending with a ‘/’ character (or appropriate path separator value depending on OS). If None, self.path is used. The final model file is typically saved to path + self.model_file_name.
• verbose (bool, default True) – Whether to log the location of the saved file.

Returns

path – Path to the saved model, minus the file name. Use this value to load the model from disk via cls.load(path), cls being the class of the model object, such as model = RFModel.load(path)

Return type

str

Module contents