Script for dictionary learning on Westgrid