1140820 meeting

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83

# libraries
library(fields)
library(SpatialTools)

# necessary functions
source("mcmc.R")
source("auxfunctions.R")

# Set up the grid
set.seed(100)
ns <- 2331
nt <- 260
s <- cbind(runif(n = ns, min = 0, max = 10), runif(n = ns, min = 0, max = 10))
x <- array(1, c(ns, nt, 3))
for (t in 1:nt) {
  x[, t, 2] <- sin(s[, 1]^(1/2))
  x[, t, 3] <- ifelse(s[, 2] > 5, cos(2 * s[, 2]^(1/2)), 1/3 * sin(s[, 2]^3))
}
plot(s)
plot(x[, 1, 2:3])

# generated data
set.seed(123)

lambda    <- c(0, 0,  0,  0,  3,  3,  3,  3,   0,   0,   0,   0,   3,   3,   3,   3,   0,   0,   0,   0,   3,   3,   3,   3)
tau.alpha <- c(0, 0,  0,  0,  6,  6,  6,  6,   0,   0,   0,   0,   6,   6,   6,   6,   0,   0,   0,   0,   6,   6,   6,   6)
tau.beta  <- c(0, 0,  0,  0, 16, 16, 16, 16,   0,   0,   0,   0,  16,  16,  16,  16,   0,   0,   0,   0,  16,  16,  16,  16)
nknots    <- c(1, 5, 10, 20,  1,  5, 10, 20,   1,   5,  10,  20,   1,   5,  10,  20,   1,   5,  10,  20,   1,   5,  10,  20)
phi.z     <- c(0, 0,  0,  0,  0,  0,  0,  0, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.9, 0.9, 0.9, 0.9, 0.9, 0.9, 0.9, 0.9)
phi.w     <- c(0, 0,  0,  0,  0,  0,  0,  0, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.9, 0.9, 0.9, 0.9, 0.9, 0.9, 0.9, 0.9)
phi.tau   <- c(0, 0,  0,  0,  0,  0,  0,  0, 0.2, 0.2, 0.2, 0.2, 0.2, 0.2, 0.2, 0.2, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5)

combined = array(NA, c(ns, nt, 24))

for (i in 1:24) {
  # --- 左右半邊不同 correlation range ---
  phi.z.local   <- ifelse(s[,1] < 5, phi.z[i],   phi.z[i] * 1.5)   # 右半邊 range 拉長
  phi.w.local   <- ifelse(s[,1] < 5, phi.w[i],   phi.w[i] * 1.5)
  phi.tau.local <- ifelse(s[,1] < 5, phi.tau[i], phi.tau[i] * 1.5)
  
  data_ <- rpotspatTS(nt = nt,
                      x = x, 
                      s = s, 
                      beta = c(10, 0, 0), 
                      gamma = 0.9, 
                      nu = 0.5, 
                      rho = 1, 
                      phi.z = phi.z[i],      # time series parameters
                      phi.w = phi.w[i],      # time series parameters
                      phi.tau = phi.tau[i],  # time series parameters
                      lambda = lambda[i], 
                      tau.alpha = tau.alpha[i], 
                      tau.beta = tau.beta[i], 
                      nknots = nknots[i], 
                      dist = "gaussian"
                      )$y
  
  # combined
  combined[, , i] = data_
  
  # hist
  hist(data_, main = paste0('hist. of ', i))
  
  # remove
  rm(data_)
}

# combine
#library(abind)
#combined <- abind(data_1, data_2, data_3, along = 3)

# save
library(dplyr)
library(reticulate)
np = import("numpy")
combined = combined %>% r_to_py()
combined = combined$transpose(2L, 1L, 0L)
save_dir = 'datasets'
if (!dir.exists(save_dir)) {
  dir.create(save_dir, recursive = TRUE)
}
np$save(paste0(save_dir, '/real.npy'), combined)
np$save(paste0(save_dir, '/all_locations.npy'), s)