Package org.apache.ignite.example.ml

Class ModelUtils

java.lang.Object

org.apache.ignite.example.ml.ModelUtils

public class ModelUtils
extends Object

Base class for ML examples with functions to 1. Download the model from DJL via URLs 2. Download the model files from any URLs 3. Deploy the Model

Constructor Summary

Constructors

Constructor	Description
ModelUtils()

Method Summary

Modifier and Type	Method	Description
static <I, O> void	downloadDjlModel(String modelUrl, Class<I> inputClass, Class<O> outputClass, String targetDirectory, String translatorFactoryClassName)	Downloads the model from DJL

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

ModelUtils

public ModelUtils()

Method Details

downloadDjlModel

public static <I,
O> void downloadDjlModel(String modelUrl,
 Class<I> inputClass,
 Class<O> outputClass,
 String targetDirectory,
 String translatorFactoryClassName)
                                   throws IOException,
ai.djl.repository.zoo.ModelNotFoundException,
ai.djl.MalformedModelException,
NoSuchMethodException,
InvocationTargetException,
InstantiationException,
IllegalAccessException,
ClassNotFoundException

Downloads the model from DJL

Type Parameters:

I - Type of Input

O - Type of Output

Parameters:

modelUrl - DJL URL of the model

inputClass - Input to the model

outputClass - Output from model

targetDirectory - Path at which the model should be downloaded

Throws:

IOException - If target directory is invalid or copy fails

ai.djl.repository.zoo.ModelNotFoundException - If model is not found for the given criteria

ai.djl.MalformedModelException - If Model files are incomplete

NoSuchMethodException

InvocationTargetException

InstantiationException

IllegalAccessException

ClassNotFoundException