Package org.gridgain.ml

Interface IgniteMl

public interface IgniteMl
Main API for machine learning inference operations in GridGain.

  • Method Details

    • predict

      <I, O> O predict(MlSimpleJobParameters<I> jobParams)
      Executes prediction using the provided model and input data synchronously.
      Type Parameters:
      I - The input type
      O - The output type
      Parameters:
      jobParams - The job parameters containing model info and input data
      Returns:
      The prediction result

    • predictAsync

      <I, O> CompletableFuture<O> predictAsync(MlSimpleJobParameters<I> jobParams)
      Executes prediction using the provided model and input data asynchronously.
      Type Parameters:
      I - The input type
      O - The output type
      Parameters:
      jobParams - The job parameters containing model info and input data
      Returns:
      A CompletableFuture containing the prediction result

    • batchPredict

      <I, O> List<O> batchPredict(MlBatchJobParameters<I> jobParams)
      Executes batch prediction using the provided model and input data synchronously.
      Type Parameters:
      I - The input type
      O - The output type
      Parameters:
      jobParams - The job parameters containing model info and batch input data
      Returns:
      List of prediction results

    • batchPredictAsync

      <I, O> CompletableFuture<List<O>> batchPredictAsync(MlBatchJobParameters<I> jobParams)
      Executes batch prediction using the provided model and input data asynchronously.
      Type Parameters:
      I - The input type
      O - The output type
      Parameters:
      jobParams - The job parameters containing model info and batch input data
      Returns:
      A CompletableFuture containing a list of prediction results

    • predictFromSql

      <O> List<O> predictFromSql(MlSqlJobParameters jobParams)
      Executes SQL-based prediction by running the query and performing inference on results. The SQL query should return the data to be used for prediction.
      Type Parameters:
      O - The output type
      Parameters:
      jobParams - The job parameters containing model info, SQL query, and parameters
      Returns:
      List of prediction results from SQL query data

    • predictFromSqlAsync

      <O> CompletableFuture<List<O>> predictFromSqlAsync(MlSqlJobParameters jobParams)
      Executes SQL-based prediction by running the query and performing inference on results asynchronously. The SQL query returns the data to be used for prediction.
      Type Parameters:
      O - The output type
      Parameters:
      jobParams - The job parameters containing model info, SQL query, and parameters
      Returns:
      A CompletableFuture containing a list of prediction results from SQL query data

    • predictColocated

      <I, O> O predictColocated(MlColocatedJobParameters<I> jobParams)
      Executes Colocated prediction by providing key and input columns to run inference on results synchronously.
      Type Parameters:
      I - The input type
      O - The output type
      Parameters:
      jobParams - The job parameters containing model info, table, key column, and the key
      Returns:
      A CompletableFuture containing the prediction result

    • predictColocatedAsync

      <I, O> CompletableFuture<O> predictColocatedAsync(MlColocatedJobParameters<I> jobParams)
      Executes Colocated prediction by providing key and input columns to run inference on results asynchronously.
      Type Parameters:
      I - The input type
      O - The output type
      Parameters:
      jobParams - The job parameters containing model info, table, key column, and the key
      Returns:
      A CompletableFuture containing the prediction result