Skip to content

Commit 0bd38cb

Browse files
committed
update tf eagerness in R package
1 parent a2d6708 commit 0bd38cb

File tree

4 files changed

+18
-4
lines changed

4 files changed

+18
-4
lines changed

R-package/DESCRIPTION

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
Package: ivis
22
Title: Artificial neural network-driven visualization of high-dimensional data using triplets.
3-
Version: 1.5.2
3+
Version: 1.5.3
44
Authors@R: c(person("Benjamin", "Szubert", email = "bszubert@beringresearch.com", role = c("aut", "cre")),
55
person("Ignat", "Drozdov", email = "idrozdov@beringresearch.com", role = c("aut")),
66
person("Kevin", "Rue-Albrecht", role = "ctb", email = "kevin.rue-albrecht@kennedy.ox.ac.uk", comment = c(ORCID = "0000-0003-3899-3872")))

R-package/R/ivis.R

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,8 @@
5656
#' be used. Otherwise, a new index will be generated and
5757
#' saved to disk in the current directory as
5858
#' 'annoy.index'.
59+
#' @param eager_execution: Whether to use eager execution with TensorFlow.
60+
# Disabled by default, as training is much faster with this option off.
5961
#' @param verbose: Controls the volume of logging output the model
6062
#' produces when training. When set to 0, silences
6163
#' outputs, when above 0 will print outputs.
@@ -74,6 +76,7 @@ ivis <- function(embedding_dims = 2L,
7476
model = "default",
7577
supervision_metric = "sparse_categorical_crossentropy",
7678
supervision_weight = 0.5,
79+
eager_execution = FALSE,
7780
annoy_index_path=NULL, verbose=1L){
7881

7982

@@ -97,6 +100,7 @@ ivis <- function(embedding_dims = 2L,
97100
supervision_metric=supervision_metric,
98101
supervision_weight=supervision_weight,
99102
annoy_index_path=annoy_index_path,
103+
eager_execution=eager_execution,
100104
verbose=verbose)
101105

102106
return(model)

R-package/man/ivis.Rd

Lines changed: 12 additions & 2 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

ivis/version.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
VERSION = '1.5.2'
1+
VERSION = '1.5.3'

0 commit comments

Comments
 (0)