Невозможно сохранить модель тензор-потока / кераса с помощью R в Rstudio на Google Cloud Platform - PullRequest
Новая
       22

Невозможно сохранить модель тензор-потока / кераса с помощью R в Rstudio на Google Cloud Platform

0 голосов
/ 21 мая 2019

Я использую Rstudio на облачной платформе Google и хочу развернуть обученную модель, но столкнулся с ошибкой.Я использую примеры, представленные на https://tensorflow.rstudio.com/keras/

Я не могу экспортировать модель в облачное хранилище Google, после чего я могу создать версию модели в облачном хранилище Google и использовать if для прогнозов.Я могу делать прогнозы в Rstudio.

Код: 

library(keras)

FLAGS <- flags(flag_numeric("dropout_rate", 0.4))

mnist <- dataset_mnist()
x_train <- mnist$train$x
y_train <- mnist$train$y
x_test <- mnist$test$x
y_test <- mnist$test$y

x_train <- array_reshape(x_train, c(nrow(x_train), 784))
x_test <- array_reshape(x_test, c(nrow(x_test), 784))
x_train <- x_train / 255
x_test <- x_test / 255

y_train <- to_categorical(y_train, 10)
y_test <- to_categorical(y_test, 10)

model <- keras_model_sequential()

model %>%
  layer_dense(units = 256, activation = 'relu', input_shape = c(784)) %>%
  layer_dropout(rate = FLAGS$dropout_rate) %>%
  layer_dense(units = 128, activation = 'relu') %>%
  layer_dropout(rate = 0.3) %>%
  layer_dense(units = 10, activation = 'softmax')

model %>% compile(
  loss = 'categorical_crossentropy',
  optimizer = optimizer_rmsprop(),
  metrics = c('accuracy')
)

model %>% fit(
  x_train, y_train,
  epochs = 20, batch_size = 128,
  validation_split = 0.2
)

model %>% evaluate(x_test, y_test)
model %>% predict_classes(x_test)

# everything works fine till above line 

export_savedmodel(model, "savedmodel") # Error line

Сообщение об ошибке: для строки в Rstudio -> export_savedmodel (модель, "сохраненная модель") 

Error in export_savedmodel.keras.engine.training.Model(model, "savedmodel") : 
      'export_savedmodel()' is currently unsupported under the TensorFlow Keras implementation, consider using 'tfestimators::keras_model_to_estimator()'

После этого я попробовал следующий код: 

#export_savedmodel(model, "savedmodel") <-- replace this line with below code
tfe_model <- tfestimators::keras_model_to_estimator(model)
export_savedmodel(tfe_model, "savedmodel")

Сообщение об ошибке: 

Error in export_savedmodel.tf_estimator(tfe_model, "savedmodel") : Currently only classifier and regressor are supported. Please specify a custom serving_input_receiver_fn.
...