Importera modeller till ditt program

I den här artikeln beskrivs import-och poängsättnings delar av ett DATABRICKS ml-modell för export . Se exportera Apache Spark ml-modeller och pipeliner för export delen av arbets flödet.

Om du vill använda modeller som exporter ATS via export av Databricks ML-modellen anropar du API: er i biblioteket dbml-local . Det här biblioteket innehåller Scala-och Java-API: er för att importera modeller och utföra poäng med låg latens (förutsägelse eller härledning).

Använda sparade modeller i Java-program

Anta att du har exporterat en Logistisk regression-pipeline och sparat den under my_models/lr_pipeline . Du kan använda ModelFactory för att skapa en LocalModel från den sparade modellens katalog och utföra en bedömning av nya data.

// Load exported model
String modelPath = "my_models/lr_pipeline";
LocalModel model = ModelFactory.loadModel(modelPath);

// The model input is a standard JSON string.
// The input schema here is: [origLabel: Double, features: Vector].
String input =
  "{\"origLabel\":-1.0," +
  "\"features\":{\"type\":0,\"size\":13," +
  "\"indices\":[0,2,3,4,6,7,8,9,10,11,12]," +
  "\"values\":[74.0,2.0,120.0,269.0,2.0,121.0,1.0,0.2,1.0,1.0,3.0]}" +
  "}";

// The model output is also a standard JSON string, with the expected output fields.
String output = model.transform(input);

Indata accepterar samma JSON-format som skapats av Apache Spark data uppsättningar och DataFrames med hjälp av Dataset.toJSON metoden (se data uppsättnings-API-dokument). Mer information finns i dbml-Local API-dokument .

Ange dbml-local biblioteks beroende i maven

Du anger dbml-local biblioteks beroendet till ett program precis som alla andra beroenden, med en maven-koordinat. Följande kodfragment ger ett exempel på hur du kan inkludera dbml-local i en maven-projekts pom.xml build-fil.

<!-- Add repository for dbml-local dependency -->
 <repositories>
   <repository>
     <snapshots>
       <enabled>false</enabled>
     </snapshots>
     <id>bintray-databricks-maven</id>
     <name>bintray</name>
     <url>https://dl.bintray.com/databricks/maven</url>
   </repository>
 </repositories>

 <dependencies>
   <!-- Main dependency for Model Scoring -->
   <dependency>
     <groupId>com.databricks</groupId>
     <artifactId>dbml-local</artifactId>
     <version>0.2.2-spark2.2</version>
   </dependency>
 </dependencies>

Laddar ned dbml-local jar v7

dbml-localJar v7 är tillgängliga från bintray.

dbml-locallicensavtalet

dbml-localBiblioteket publiceras under MIT-licensen.

Exempel program

Du kan visa ett väldigt enkelt exempel program som visar hur du använder export bidrags biblioteket Databricks ML dbml-local -modell i Databricks-ml-GitHub-lagringsplatsen. Den här demon innehåller Databricks-anteckningsböcker för utbildning och export av MLlib-modeller. Dessa antecknings böcker är kopplade till enkla Java-program som visar hur du importerar modeller och gör förutsägelser.