Uploaded image for project: 'SystemDS'
  1. SystemDS
  2. SYSTEMDS-2505

Support generation of DML script using Keras2DML and Caffe2DML

    XMLWordPrintableJSON

Details

    • Test
    • Status: Resolved
    • Major
    • Resolution: Fixed
    • None
    • SystemML 1.3
    • APIs
    • None

    Description

       

      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())
      

      Attachments

        Activity

          People

            niketanpansare Niketan Pansare
            niketanpansare Niketan Pansare
            Votes:
            0 Vote for this issue
            Watchers:
            1 Start watching this issue

            Dates

              Created:
              Updated:
              Resolved: