trainstation has been developed at the Department of Physics of Chalmers University of Technology in Gothenburg, Sweden with funding from the Knut och Alice Wallenbergs Foundation, the Swedish Research Council, the Swedish Foundation for Strategic Research, and the Swedish National Infrastructure for Computing. Internally it heavily relies on scikit-learn. When using trainstation in your research please therefore cite the following two papers:
Scikit-Learn: Machine Learning in Python
Fabian Pedregosa et al.
Journal of Machine Learning Research 12, 2825 (2011)
Efficient construction of linear models in materials modeling and applications to force constant expansions
Erik Fransson, Fredrik Eriksson, and Paul Erhart
npj Computational Materials 6, 135 (2020)