rail.core.algo_utils module

Utility functions to test alogrithms

rail.core.algo_utils.one_algo(key, single_trainer, single_estimator, train_kwargs, estim_kwargs)[source]

A basic test of running an estimator subclass Run inform, write temporary trained model to ‘tempmodelfile.tmp’, run photo-z algorithm. Then, load temp modelfile and re-run, return both datasets.