Added weight map to SampleDataGenerator