Skip to main content

PyTorchUtils

C++: phynexis::ml::PyTorchUtils
Python: phynexis.ml.PyTorchUtils
Header: src/ml/pytorch_interface.hpp

Description

Utility functions for PyTorch interoperability.

Methods

get_labels(...)

Converts model output to class labels.

get_mae(...)

Computes mean absolute error.

get_mse(...)

Computes mean squared error.