Merge pull request #681 from kevinykuo/bugfix/tfkeras-backend

Fix k_set_epsilon() and k_set_floatx() for tensorflow

Author: Sigrid Keydana

DESCRIPTION

@@ -1,7 +1,7 @@
 Package: keras
 Type: Package
 Title: R Interface to 'Keras'
-Version: 2.2.4.9000
+Version: 2.2.4.9001
 Authors@R: c(
   person("JJ", "Allaire", role = c("aut", "cre"), email = "[email protected]"),
   person("François", "Chollet", role = c("aut", "cph")),

R/backend.R

@@ -867,9 +867,7 @@ k_epsilon <- function() {
 #' @rdname k_epsilon
 #' @export
 k_set_epsilon <- function(e) {
-  keras$backend$set_epsilon(
-    e = e
-  )
+  keras$backend$set_epsilon(e)
 }
 
 
@@ -994,9 +992,7 @@ k_floatx <- function() {
 #' @rdname k_floatx
 #' @export
 k_set_floatx <- function(floatx) {
-  keras$backend$set_floatx(
-    floatx = floatx
-  )
+  keras$backend$set_floatx(floatx)
 }
 
 

tests/testthat/test-backend.R

@@ -115,3 +115,20 @@ test_backend("k_concatenate", {
   k_concatenate(list(x, y), 1)
 })
 
+test_backend("k_set_floatx", {
+  skip_if_cntk()
+  skip_if_theano()
+  k_set_floatx("float16")
+  expect_identical(k_floatx(), "float16")
+  k_set_floatx("float32")
+  expect_identical(k_floatx(), "float32")
+})
+
+test_backend("k_set_epsilon", {
+  skip_if_cntk()
+  skip_if_theano()
+  k_set_epsilon(1e-4)
+  expect_equal(k_epsilon(), 1e-4)
+  k_set_epsilon(1e-07)
+  expect_equal(k_epsilon(), 1e-07)
+})