Search in sources :

Example 91 with Script

use of org.apache.sysml.api.mlcontext.Script in project incubator-systemml by apache.

the class MLContextTest method testOutputListStringCSVSparseDML.

@Test
public void testOutputListStringCSVSparseDML() {
    System.out.println("MLContextTest - output List String CSV Sparse DML");
    String s = "M = matrix(0, rows=10, cols=10); M[1,1]=1; M[1,2]=2; M[2,1]=3; M[2,2]=4; print(toString(M));";
    Script script = dml(s).out("M");
    MLResults results = ml.execute(script);
    MatrixObject mo = results.getMatrixObject("M");
    List<String> lines = MLContextConversionUtil.matrixObjectToListStringCSV(mo);
    Assert.assertEquals("1.0,2.0", lines.get(0));
    Assert.assertEquals("3.0,4.0", lines.get(1));
}
Also used : Script(org.apache.sysml.api.mlcontext.Script) MatrixObject(org.apache.sysml.runtime.controlprogram.caching.MatrixObject) MLResults(org.apache.sysml.api.mlcontext.MLResults) Test(org.junit.Test)

Example 92 with Script

use of org.apache.sysml.api.mlcontext.Script in project incubator-systemml by apache.

the class MLContextTest method testGetTuple2DML.

@Test
public void testGetTuple2DML() {
    System.out.println("MLContextTest - Get Tuple2<Matrix,Double> DML");
    double[][] m = new double[][] { { 1, 2 }, { 3, 4 } };
    Script script = dml("N=M*2;s=sum(N)").in("M", m).out("N", "s");
    Tuple2<Matrix, Double> tuple = ml.execute(script).getTuple("N", "s");
    double[][] n = tuple._1().to2DDoubleArray();
    double s = tuple._2();
    Assert.assertArrayEquals(new double[] { 2, 4 }, n[0], 0.0);
    Assert.assertArrayEquals(new double[] { 6, 8 }, n[1], 0.0);
    Assert.assertEquals(20.0, s, 0.0);
}
Also used : Script(org.apache.sysml.api.mlcontext.Script) Matrix(org.apache.sysml.api.mlcontext.Matrix) Test(org.junit.Test)

Example 93 with Script

use of org.apache.sysml.api.mlcontext.Script in project incubator-systemml by apache.

the class MLContextTest method testCSVMatrixFromURLSumPYDML.

@Test
public void testCSVMatrixFromURLSumPYDML() throws MalformedURLException {
    System.out.println("MLContextTest - CSV matrix from URL sum PYDML");
    String csv = "https://raw.githubusercontent.com/apache/systemml/master/src/test/scripts/org/apache/sysml/api/mlcontext/1234.csv";
    URL url = new URL(csv);
    Script script = pydml("print('sum: ' + sum(M))").in("M", url);
    setExpectedStdOut("sum: 10.0");
    ml.execute(script);
}
Also used : Script(org.apache.sysml.api.mlcontext.Script) URL(java.net.URL) Test(org.junit.Test)

Example 94 with Script

use of org.apache.sysml.api.mlcontext.Script in project incubator-systemml by apache.

the class MLContextTest method testPrintFormattingBooleanSubstitutionVariables.

@Test
public void testPrintFormattingBooleanSubstitutionVariables() {
    System.out.println("MLContextTest - print formatting boolean substitution variables");
    Script script = dml("a=TRUE; b=FALSE; print('%b %b', a, b);");
    setExpectedStdOut("true false");
    ml.execute(script);
}
Also used : Script(org.apache.sysml.api.mlcontext.Script) Test(org.junit.Test)

Example 95 with Script

use of org.apache.sysml.api.mlcontext.Script in project incubator-systemml by apache.

the class MLContextTest method testDisplayDoubleAddDML.

@Test
public void testDisplayDoubleAddDML() {
    System.out.println("MLContextTest - display double add DML");
    String s = "print(i+j);";
    Script script = dml(s).in("i", 5.1).in("j", 6.2);
    setExpectedStdOut("11.3");
    ml.execute(script);
}
Also used : Script(org.apache.sysml.api.mlcontext.Script) Test(org.junit.Test)

Aggregations

Script (org.apache.sysml.api.mlcontext.Script)206 Test (org.junit.Test)188 ArrayList (java.util.ArrayList)64 Row (org.apache.spark.sql.Row)53 MLResults (org.apache.sysml.api.mlcontext.MLResults)47 StructType (org.apache.spark.sql.types.StructType)36 StructField (org.apache.spark.sql.types.StructField)35 MatrixMetadata (org.apache.sysml.api.mlcontext.MatrixMetadata)35 DenseVector (org.apache.spark.ml.linalg.DenseVector)14 Vector (org.apache.spark.ml.linalg.Vector)14 VectorUDT (org.apache.spark.ml.linalg.VectorUDT)14 Matrix (org.apache.sysml.api.mlcontext.Matrix)13 MatrixObject (org.apache.sysml.runtime.controlprogram.caching.MatrixObject)13 MLContext (org.apache.sysml.api.mlcontext.MLContext)9 MatrixBlock (org.apache.sysml.runtime.matrix.data.MatrixBlock)9 Tuple2 (scala.Tuple2)9 List (java.util.List)8 URL (java.net.URL)7 DMLScript (org.apache.sysml.api.DMLScript)7 Seq (scala.collection.Seq)7