geoprior.models.subsidence.batch_io#

Batch.io

Functions

select_q(pred[, quantiles, q, fallback])

Select q-quantile from pred.

tile_true_to_quantiles(y_true, y_pred)

Make y_true compatible with y_pred when y_pred has quantile axis.
geoprior.models.subsidence.batch_io.select_q(pred, quantiles=None, q=0.5, fallback='mean')[source]#

Select q-quantile from pred.

pred:

  • (B,H,Q,O) -> returns (B,H,O)

  • (B,H,Q,1) -> returns (B,H,1)

  • otherwise returned as-is.

geoprior.models.subsidence.batch_io.tile_true_to_quantiles(y_true, y_pred)[source]#

Make y_true compatible with y_pred when y_pred has quantile axis.

y_true: (B,H,O) and y_pred: (B,H,Q,O) -> return y_true_q: (B,H,Q,O) Else return y_true unchanged.