Sample Class

Sample n rows for each value of a column

Parameters

Name	Type	Description	Is Optional
sample_col	column	The column for which you want to sample
sample	dict	Value of column as a key, n rows to be sampled as values

Example

ds = rasgo.get.dataset(id)

ds2 = ds.sample_class(sample_col='BINARY_TARGET_COLUMNNAME', sample={'1':15000, '0':60000})
ds2.preview()

Source Code

RasgoTransforms/rasgotransforms/rasgotransforms/transforms/sample_class/sample_class.sql at main · rasgointelligence/RasgoTransformsGitHub