ise.models.density_estimators package

Submodules

ise.models.density_estimators.normalizing_flow module

class ise.models.density_estimators.normalizing_flow.NormalizingFlow(input_size=43, output_size=1, output_sequence_length=86, num_flow_transforms=5)

Bases: Module

A class representing a Normalizing Flow model.

aleatoric(features, num_samples, batch_size=128)

fit(X, y, epochs=100, batch_size=64, save_checkpoints=True, checkpoint_path='checkpoint.pt', early_stopping=True, patience=10, verbose=True)

get_latent(x, latent_constant=0.0)

static load(path)

sample(features, num_samples, return_type='numpy')

save(path)

Module contents