Regression capability in geowombat about geowombat HOT 3 CLOSED

with gw.config.update(ref_image=target_string):
    with gw.open(select_images, nodata=9999, stack_dim="band") as src:

          # Create a prediction stack where each variable used in regression is a band
          src.gw.save(
              "outputs/pred_stack.tif",
              compress="lzw",
              overwrite=True,  # bigtiff=True
          )
        # extract variables
        df = gw.extract(src, lu_poly, verbose=1)
        y = something
        X = something

# Create a pipeline with the preprocessor and the OLS model
pipeline_performance = Pipeline([
    ('scaler', StandardScaler()),
    ('model', LinearRegression())
])

# use sklearn to fit desired model
pipeline_performance.fit(X, y)

# predict to stack
def user_func(w, block, model):
    pred_shape = list(block.shape)
    X = block.reshape(pred_shape[0], -1).T
    pred_shape[0] = 1
    y_hat = model.predict(X)
    X_reshaped = y_hat.T.reshape(pred_shape)
    return w, X_reshaped

# create prediction image
gw.apply(
    "outputs/pred_stack.tif",
    f"outputs/final_model_OLS_prediction.tif",
    user_func,
    args=(pipeline_performance,),
    n_jobs=16,
    count=1,
)