Skip to contents

Extract the number of neighbors for a PrestoGP model.

get_neighbors-PrestoGPModel-method.Rd

This method extracts the number of nearest neighbors that were used to compute the PrestoGP model.

Usage

# S4 method for class 'PrestoGPModel'
get_neighbors(model)

Arguments

model

The PrestoGP model object

Value

The (integer) number of neighbors used to fit the model.

References

  • Messier, K.P. and Katzfuss, M. "Scalable penalized spatiotemporal land-use regression for ground-level nitrogen dioxide", The Annals of Applied Statistics (2021) 15(2):688-710.

See also

PrestoGPModel-class, prestogp_fit

Examples

data(soil)
soil <- soil[!is.na(soil[,5]),] # remove rows with NA's
y <- soil[,4]                   # predict moisture content
X <- as.matrix(soil[,5:9])
locs <- as.matrix(soil[,1:2])

soil.vm <- new("VecchiaModel", n_neighbors = 10)
soil.vm <- prestogp_fit(soil.vm, y, X, locs)
#> 
#> Beginning iteration 1 
#> Estimating theta... 
#> Estimation of theta complete 
#> Estimating beta... 
#> Estimation of beta complete 
#> Iteration 1 complete 
#> Current penalized likelihood: 487.3414 
#> Current MSE: 9.104869 
#> Beginning iteration 2 
#> Estimating theta... 
#> Estimation of theta complete 
#> Estimating beta... 
#> Estimation of beta complete 
#> Iteration 2 complete 
#> Current penalized likelihood: 487.2106 
#> Current MSE: 9.107132 
#> Beginning iteration 3 
#> Estimating theta... 
#> Estimation of theta complete 
#> Estimating beta... 
#> Estimation of beta complete 
#> Iteration 3 complete 
#> Current penalized likelihood: 487.2106 
#> Current MSE: 9.10711 
get_neighbors(soil.vm)
#> [1] 10