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.