use of org.apache.sysml.api.mlcontext.Script in project incubator-systemml by apache.
the class MLContextTest method testDataFrameSumPYDMLVectorWithNoIDColumn.
@Test
public void testDataFrameSumPYDMLVectorWithNoIDColumn() {
System.out.println("MLContextTest - DataFrame sum PYDML, vector with no ID column");
List<Vector> list = new ArrayList<Vector>();
list.add(Vectors.dense(1.0, 2.0, 3.0));
list.add(Vectors.dense(4.0, 5.0, 6.0));
list.add(Vectors.dense(7.0, 8.0, 9.0));
JavaRDD<Vector> javaRddVector = sc.parallelize(list);
JavaRDD<Row> javaRddRow = javaRddVector.map(new VectorRow());
List<StructField> fields = new ArrayList<StructField>();
fields.add(DataTypes.createStructField("C1", new VectorUDT(), true));
StructType schema = DataTypes.createStructType(fields);
Dataset<Row> dataFrame = spark.createDataFrame(javaRddRow, schema);
MatrixMetadata mm = new MatrixMetadata(MatrixFormat.DF_VECTOR);
Script script = pydml("print('sum: ' + sum(M))").in("M", dataFrame, mm);
setExpectedStdOut("sum: 45.0");
ml.execute(script);
}
use of org.apache.sysml.api.mlcontext.Script in project incubator-systemml by apache.
the class MLContextTest method testPrintFormattingStringSubstitution.
@Test
public void testPrintFormattingStringSubstitution() {
System.out.println("MLContextTest - print formatting string substitution");
Script script = dml("print('hello %s', 'world');");
setExpectedStdOut("hello world");
ml.execute(script);
}
use of org.apache.sysml.api.mlcontext.Script in project incubator-systemml by apache.
the class MLContextTest method testFunctionTwoReturnValuesPYDML.
@Test
public void testFunctionTwoReturnValuesPYDML() {
System.out.println("MLContextTest - function with two return values PYDML");
String s = "def hello()->(s1:str,s2:str):\n\ts1='return'\n\ts2='values'\n[a,b]=hello()\nprint(a+' '+b)";
Script script = pydml(s);
setExpectedStdOut("return values");
ml.execute(script);
}
use of org.apache.sysml.api.mlcontext.Script in project incubator-systemml by apache.
the class MLContextTest method testGetTuple4DML.
@Test
public void testGetTuple4DML() {
System.out.println("MLContextTest - Get Tuple4<Long,Double,Boolean,String> DML");
Script script = dml("a=1+2;b=a+0.5;c=TRUE;d=\"yes it's \"+c").out("a", "b", "c", "d");
Tuple4<Long, Double, Boolean, String> tuple = ml.execute(script).getTuple("a", "b", "c", "d");
long a = tuple._1();
double b = tuple._2();
boolean c = tuple._3();
String d = tuple._4();
Assert.assertEquals(3, a);
Assert.assertEquals(3.5, b, 0.0);
Assert.assertEquals(true, c);
Assert.assertEquals("yes it's TRUE", d);
}
use of org.apache.sysml.api.mlcontext.Script in project incubator-systemml by apache.
the class MLContextTest method testDisplayStringConcatenationPYDML.
@Test
public void testDisplayStringConcatenationPYDML() {
System.out.println("MLContextTest - display string concatenation PYDML");
String s = "print(str1+str2)";
Script script = pydml(s).in("str1", "hello").in("str2", "goodbye");
setExpectedStdOut("hellogoodbye");
ml.execute(script);
}
Aggregations