Currently running the model on many millions of observations is computationally costly.
A better solution would be to run the model a sample of observations (E.g., 10K-50K observations for sample and target). And then export the model, and use it to predict the weights of other observations.
This should probably be done only after issue #30 is resolved.