Search in sources :

Example 6 with Seq

use of scala.collection.Seq in project incubator-systemml by apache.

the class MLContextTest method testInputTupleSeqWithMetadataPYDML.

@SuppressWarnings({ "rawtypes", "unchecked" })
@Test
public void testInputTupleSeqWithMetadataPYDML() {
    System.out.println("MLContextTest - Tuple sequence with metadata PYDML");
    List<String> list1 = new ArrayList<String>();
    list1.add("1,2");
    list1.add("3,4");
    JavaRDD<String> javaRDD1 = sc.parallelize(list1);
    RDD<String> rdd1 = JavaRDD.toRDD(javaRDD1);
    List<String> list2 = new ArrayList<String>();
    list2.add("5,6");
    list2.add("7,8");
    JavaRDD<String> javaRDD2 = sc.parallelize(list2);
    RDD<String> rdd2 = JavaRDD.toRDD(javaRDD2);
    MatrixMetadata mm1 = new MatrixMetadata(2, 2);
    MatrixMetadata mm2 = new MatrixMetadata(2, 2);
    Tuple3 tuple1 = new Tuple3("m1", rdd1, mm1);
    Tuple3 tuple2 = new Tuple3("m2", rdd2, mm2);
    List tupleList = new ArrayList();
    tupleList.add(tuple1);
    tupleList.add(tuple2);
    Seq seq = JavaConversions.asScalaBuffer(tupleList).toSeq();
    Script script = pydml("print('sums: ' + sum(m1) + ' ' + sum(m2))").in(seq);
    setExpectedStdOut("sums: 10.0 26.0");
    ml.execute(script);
}
Also used : Script(org.apache.sysml.api.mlcontext.Script) Tuple3(scala.Tuple3) ArrayList(java.util.ArrayList) List(java.util.List) ArrayList(java.util.ArrayList) MatrixMetadata(org.apache.sysml.api.mlcontext.MatrixMetadata) Seq(scala.collection.Seq) Test(org.junit.Test)

Example 7 with Seq

use of scala.collection.Seq in project incubator-systemml by apache.

the class MLContextTest method testOutputScalaSeqDML.

@SuppressWarnings({ "unchecked", "rawtypes" })
@Test
public void testOutputScalaSeqDML() {
    System.out.println("MLContextTest - output specified as Scala Seq DML");
    List outputs = Arrays.asList("x", "y");
    Seq seq = JavaConversions.asScalaBuffer(outputs).toSeq();
    Script script = dml("a=1;x=a+1;y=x+1").out(seq);
    MLResults results = ml.execute(script);
    Assert.assertEquals(2, results.getLong("x"));
    Assert.assertEquals(3, results.getLong("y"));
}
Also used : Script(org.apache.sysml.api.mlcontext.Script) MLResults(org.apache.sysml.api.mlcontext.MLResults) List(java.util.List) ArrayList(java.util.ArrayList) Seq(scala.collection.Seq) Test(org.junit.Test)

Aggregations

ArrayList (java.util.ArrayList)7 List (java.util.List)7 Seq (scala.collection.Seq)7 Script (org.apache.sysml.api.mlcontext.Script)6 Test (org.junit.Test)6 Tuple2 (scala.Tuple2)3 Tuple3 (scala.Tuple3)3 MLResults (org.apache.sysml.api.mlcontext.MLResults)2 MatrixMetadata (org.apache.sysml.api.mlcontext.MatrixMetadata)2 AbstractIterator (com.google.common.collect.AbstractIterator)1 Collection (java.util.Collection)1 OutputWindowedValue (org.apache.beam.runners.core.OutputWindowedValue)1 ReduceFnRunner (org.apache.beam.runners.core.ReduceFnRunner)1 SystemReduceFn (org.apache.beam.runners.core.SystemReduceFn)1 TimerInternals (org.apache.beam.runners.core.TimerInternals)1 UnsupportedSideInputReader (org.apache.beam.runners.core.UnsupportedSideInputReader)1 CounterCell (org.apache.beam.runners.core.metrics.CounterCell)1 MetricsContainerImpl (org.apache.beam.runners.core.metrics.MetricsContainerImpl)1 SparkPipelineOptions (org.apache.beam.runners.spark.SparkPipelineOptions)1 ByteArray (org.apache.beam.runners.spark.util.ByteArray)1