Merge pull request #995 from dfalbel/get_vocabulary

get_vocabulary performance

Author: dfalbel

R/layer-text_vectorization.R

@@ -93,7 +93,9 @@ layer_text_vectorization <- function(object, max_tokens = NULL, standardize = "l
 #' @seealso [set_vocabulary()]
 #' @export
 get_vocabulary <- function(object) {
-  object$get_vocabulary()
+  python_path <- system.file("python", package = "keras")
+  tools <- import_from_path("kerastools", path = python_path)
+  tools$get_vocabulary$get_vocabulary(object)
 }
 
 #' Sets vocabulary (and optionally document frequency) data for the layer

inst/python/kerastools/get_vocabulary.py

@@ -0,0 +1,5 @@
+
+def get_vocabulary (layer):
+  vocab = layer.get_vocabulary()
+  return([x.decode("UTF-8") for x in vocab])
+