| args <- commandArgs(trailingOnly = FALSE) |
| script_arg <- grep("^--file=", args, value = TRUE) |
| if (length(script_arg) > 0) { |
| script_path <- normalizePath(sub("^--file=", "", script_arg[1])) |
| setwd(normalizePath(file.path(dirname(script_path), ".."))) |
| } |
|
|
| library(reticulate) |
| use_python(Sys.getenv("RETICULATE_PYTHON"), required = TRUE) |
| np <- import("numpy", convert = FALSE) |
|
|
| save_npz <- function(out_file, x_train, y_train, x_test, y_test) { |
| np$savez_compressed( |
| out_file, |
| x_train = np$array(readRDS(x_train), dtype = "float32"), |
| y_train = np$array(readRDS(y_train), dtype = "float32"), |
| x_test = np$array(readRDS(x_test), dtype = "float32"), |
| y_test = np$array(readRDS(y_test), dtype = "float32") |
| ) |
| } |
|
|
| save_npz("model/CNN/c1_data.npz", "model/CNN/c1_train.RDS", "model/CNN/c1_train_y.RDS", "model/CNN/c1_test.RDS", "model/CNN/c1_test_y.RDS") |
| save_npz("model/CNN/p1_data.npz", "model/CNN/p1_train.RDS", "model/CNN/p1_train_y.RDS", "model/CNN/p1_test.RDS", "model/CNN/p1_test_y.RDS") |
|
|
| cat("export cnn npz OK\n") |
|
|
