use of org.apache.sysml.api.mlcontext.Script in project incubator-systemml by apache.
the class MLContextTest method testPrintFormattingIntegerSubstitutionAlignment.
@Test
public void testPrintFormattingIntegerSubstitutionAlignment() {
System.out.println("MLContextTest - print formatting integer substitution alignment");
Script script = dml("print(\"'%10d' '%-10d'\", 42, 43);");
setExpectedStdOut("' 42' '43 '");
ml.execute(script);
}
use of org.apache.sysml.api.mlcontext.Script in project incubator-systemml by apache.
the class MLContextTest method testCustomExecutionStepPYDML.
@Test
public void testCustomExecutionStepPYDML() {
System.out.println("MLContextTest - custom execution step PYDML");
String testString = "custom execution step";
setExpectedStdOut(testString);
Script script = new Script("print('" + testString + "')", org.apache.sysml.api.mlcontext.ScriptType.PYDML);
ScriptExecutor scriptExecutor = new ScriptExecutor() {
@Override
protected void showExplanation() {
}
};
ml.execute(script, scriptExecutor);
}
use of org.apache.sysml.api.mlcontext.Script in project incubator-systemml by apache.
the class MLContextTest method testRDDGoodMetadataDML.
@Test
public void testRDDGoodMetadataDML() {
System.out.println("MLContextTest - RDD<String> good metadata DML");
List<String> list = new ArrayList<String>();
list.add("1,1,1");
list.add("2,2,2");
list.add("3,3,3");
JavaRDD<String> javaRDD = sc.parallelize(list);
RDD<String> rdd = JavaRDD.toRDD(javaRDD);
MatrixMetadata mm = new MatrixMetadata(3, 3, 9);
Script script = dml("print('sum: ' + sum(M));").in("M", rdd, mm);
setExpectedStdOut("sum: 18.0");
ml.execute(script);
}
use of org.apache.sysml.api.mlcontext.Script in project incubator-systemml by apache.
the class MLContextTest method testOutputListDML.
@Test
public void testOutputListDML() {
System.out.println("MLContextTest - output specified as List DML");
List<String> outputs = Arrays.asList("x", "y");
Script script = dml("a=1;x=a+1;y=x+1").out(outputs);
MLResults results = ml.execute(script);
Assert.assertEquals(2, results.getLong("x"));
Assert.assertEquals(3, results.getLong("y"));
}
use of org.apache.sysml.api.mlcontext.Script in project incubator-systemml by apache.
the class MLContextTest method testOutputDataFrameDMLVectorWithIDColumn.
@Test
public void testOutputDataFrameDMLVectorWithIDColumn() {
System.out.println("MLContextTest - output DataFrame DML, vector with ID column");
String s = "M = matrix('1 2 3 4', rows=2, cols=2);";
Script script = dml(s).out("M");
MLResults results = ml.execute(script);
Dataset<Row> dataFrame = results.getDataFrameVectorWithIDColumn("M");
List<Row> list = dataFrame.collectAsList();
Row row1 = list.get(0);
Assert.assertEquals(1.0, row1.getDouble(0), 0.0);
Assert.assertArrayEquals(new double[] { 1.0, 2.0 }, ((Vector) row1.get(1)).toArray(), 0.0);
Row row2 = list.get(1);
Assert.assertEquals(2.0, row2.getDouble(0), 0.0);
Assert.assertArrayEquals(new double[] { 3.0, 4.0 }, ((Vector) row2.get(1)).toArray(), 0.0);
}
Aggregations