systemml-issues mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From "Niketan Pansare (JIRA)" <j...@apache.org>
Subject [jira] [Created] (SYSTEMML-2505) Support generation of DML script using Keras2DML and Caffe2DML
Date Sat, 08 Dec 2018 19:19:00 GMT
Niketan Pansare created SYSTEMML-2505:
-----------------------------------------

             Summary: Support generation of DML script using Keras2DML and Caffe2DML
                 Key: SYSTEMML-2505
                 URL: https://issues.apache.org/jira/browse/SYSTEMML-2505
             Project: SystemML
          Issue Type: Test
          Components: APIs
            Reporter: Niketan Pansare
            Assignee: Niketan Pansare
             Fix For: SystemML 1.3


 
{code}
from keras.applications.vgg16 import VGG16

keras_model = VGG16(weights="imagenet", pooling="max") 

from systemml.mllearn import Keras2DML

sysml_model = Keras2DML(spark, keras_model, input_shape=(3,224,224), weights='weights_dir')

sysml_model.set(test_algo='batch', train_algo='minibatch')

print(sysml_model.get_training_script())

print(sysml_model.get_prediction_script())
{code}



--
This message was sent by Atlassian JIRA
(v7.6.3#76005)

Mime
View raw message