2017-05-18 4 views
0

Haben Sie eine trainierte DRFModel-Instanz in scala, was ist der beste Weg, das entsprechende MojoModel-Objekt für das Scoring zu generieren? Von den APIs, die ich bisher gesehen habe, geht es meist darum, in eine Datei zu exportieren und dann zB mit dem MojoModel.load (Pfad) zu laden, aber keine direkte Konvertierung?H2O Mojo-Modell von DRFModel

Antwort

1

Die Modellinstanz kann derzeit nicht in eine Mojo-Instanz konvertiert werden, ohne MojoWriter zu durchlaufen.

MojoWriter bietet Methode

abstract public void writeTo(OutputStream os); 

Sie können damit die mojo auf ein Byte-Array schreiben (unter Verwendung eines ByteArrayOutputStream) und dann als eine Quelle des mojo Daten verwenden:

ByteArrayOutputStream os = new ByteArrayOutputStream(); 
    model.getMojo().writeTo(os); 
    MojoModel mojoModel = MojoModel.load(MojoReaderBackendFactory.createReaderBackend(
      new ByteArrayInputStream(os.toByteArray()), MojoReaderBackendFactory.CachingStrategy.MEMORY));