Create dataset generator