mdtraj and sklearn