diff --git a/DESCRIPTION b/DESCRIPTION index 1c922d32..c69c4490 100644 --- a/DESCRIPTION +++ b/DESCRIPTION @@ -7,8 +7,9 @@ Authors@R: c( person("Kevin", "Ushey", role = c("aut")), person(family = "RStudio", role = c("cph")) ) -Description: Interface to the Google Cloud Machine Learning Platform which provides - cloud tools for training machine learning models. +Description: Interface to the Google Cloud Machine Learning Platform + , which provides cloud tools for training machine + learning models. Depends: R (>= 3.3.0), tfruns (>= 1.3) diff --git a/vignettes/storage.Rmd b/vignettes/storage.Rmd index 6edca593..cab8979b 100644 --- a/vignettes/storage.Rmd +++ b/vignettes/storage.Rmd @@ -77,7 +77,7 @@ train_generator <- flow_images_from_directory( gs_local_dir("gs://quarter-deck-529/images/train"), image_data_generator(rescale = 1/255), target_size = c(150, 150), - batch_size = 32 + batch_size = 32, class_mode = "binary" ) ```