I have achieved this using sklearn - I just pass the object name as a string in the parameters.yml and then I parse it with getattr() to get the specific object