morpheus.messages.multi_response_message.ResponseMemoryAE

class ResponseMemoryAE(*args, **kwargs)[source]

Bases: morpheus.messages.multi_response_message.ResponseMemory

Subclass of ResponseMemory specific to the AutoEncoder pipeline.

Parameters
probscupy.ndarray

Probabilities tensor

user_idstr

User id the inference was performed against.

explain_dfpd.Dataframe

Explainability Dataframe, for each feature a column will exist with a name in the form of: {feature}_z_loss containing the loss z-score along with max_abs_z and mean_abs_z columns

Attributes
explain_df
probs

Methods

get_output(name)

Return the output tensor specified by name.

get_output(name)[source]

Return the output tensor specified by name.

Parameters
namestr

Name of output tensor.

Returns
cupy.ndarray

Tensor corresponding to name.

© Copyright 2023, NVIDIA. Last updated on Feb 3, 2023.