Removing default environment from Scala DSL

AlicTsai · Apr 3, 2014 · d583f14 · d583f14
1 parent fcc23d7
commit d583f14
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 16 deletions.
diff --git a/h2o-scala/src/main/scala/water/api/dsl/DefaultEnv.scala b/h2o-scala/src/main/scala/water/api/dsl/DefaultEnv.scala
diff --git a/h2o-scala/src/main/scala/water/api/dsl/H2ODsl.scala b/h2o-scala/src/main/scala/water/api/dsl/H2ODsl.scala
@@ -5,7 +5,7 @@ import water.fvec.Frame
 import water.Key
 
 /** The object carry global environment and provides basic global methods such as head, tail, nrows, ... */
-object H2ODsl extends H2ODslImplicitConv with T_R_Env[DFrame] with T_H2O_Env[HexKey, DFrame] with DefaultEnv[HexKey, DFrame] {
+object H2ODsl extends H2ODslImplicitConv with T_R_Env[DFrame] with T_H2O_Env[HexKey, DFrame] {
   // Binary operator type alias
   type BOp = T_NV_Transf[scala.Double]
   // Filter operator type alias

diff --git a/h2o-scala/src/main/scala/water/api/dsl/examples/Examples.scala b/h2o-scala/src/main/scala/water/api/dsl/examples/Examples.scala
@@ -57,9 +57,9 @@ makes a prediction over train data and compute MSE of prediction."""")
     val predict:DFrame = model.score(f.frame())
 
     println("Prediction on train data: \n" + predict)
-    
-    // compute mean squared errors
-    val serr = (response - predict)^2
+
+    val response = f(2) // take 2nd column as response
+    val serr = (response - predict)^2 // compute mean squared errors
     println("Errors per row: " + serr)
     // make a sum
     val rss = serr collect (0.0, new CDOp() {