def assign_record_to_split()

in 06_preprocessing/jpeg_to_tfrecord_tft.py [0:0]

• 7 lines of code
• 3 McCabe index (conditional complexity)

def assign_record_to_split(rec):
    rnd = np.random.rand()
    if rnd < 0.8:
        return ('train', rec)
    if rnd < 0.9:
        return ('valid', rec)
    return ('test', rec)