search

aangelopoulos / ppi_py

A package for statistically rigorous scientific discovery using machine learning. Implements prediction-powered inference.
MIT License
194 stars 14 forks source link

`ppi_mean_ci` does not support multidimensional inference problems #6

Closed PierreBoyeau closed 6 months ago

PierreBoyeau commented 7 months ago

The ppi_mean_ci function does not handle well cases where labelled/unlabelled datapoints are multidimensional.

The observation weights do not broadcast well with the Y matrix. For instance, here, weights of shape (n_obs,), can't be multiplied with Y, when the latter are of shape (n_obs, d).