ForestFlow icon indicating copy to clipboard operation
ForestFlow copied to clipboard

Add ONNX flavor support

Open aalkilani opened this issue 5 years ago • 2 comments

Currently ForestFlow only supports H2O flavors. This is extensible and we'd like to add support for ONNX-based models to cover a wider variety of frameworks that know how to export to ONNX.

This SHOULD look into runtime/platform considerations and provide switches for GPU support for example.

aalkilani avatar Jun 05 '20 00:06 aalkilani

+1

You may find my project ONNX-Scala useful here.

EmergentOrder avatar Jan 16 '21 20:01 EmergentOrder

Ahh, I see MLflow has ONNX support now, however it seems to be limited: https://github.com/mlflow/mlflow/issues/3229

EmergentOrder avatar Jan 16 '21 20:01 EmergentOrder