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

Support generation of DML script using Keras2DML and Caffe2DML

    XMLWordPrintableJSON

    Details

    • Type: Test
    • Status: Resolved
    • Priority: Major
    • Resolution: Fixed
    • Affects Version/s: None
    • Fix Version/s: SystemML 1.3
    • Component/s: APIs
    • Labels:
      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

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

              Dates

              • Created:
                Updated:
                Resolved: