stpredictions.models.IOKR package

Submodules

stpredictions.models.IOKR.model module

class stpredictions.models.IOKR.model.IOKR

Bases: object

Class used to apply Input and Output Kernel Regression

alpha(X_test)

fit(X, Y, L, input_kernel='linear', input_kernel_param=None): Model Fitting

predict(X_test, Y_candidates, output_kernel='linear', output_kernel_param=None): Model Prediction

stpredictions.models.IOKR.utils module

class stpredictions.models.IOKR.utils.MyDataset(inputs, labels): Bases: torch.utils.data.dataset.Dataset

class stpredictions.models.IOKR.utils.SGD(params, lr=0.5, momentum=0.0, weight_decay=0.0)

Bases: object

update(gradients)

stpredictions.models.IOKR.utils.create_path_that_doesnt_exist(path_save: str, file_name: str, extension: str)

stpredictions.models.IOKR.utils.project_root() → pathlib.Path: Returns project root folder.

Module contents