Dataset Clustering for Improved Offline Policy Learning