def create_train_dataset()

in courses/DSL/challenge-mlprep/fraud_detection/trainer/model.py [0:0]

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

def create_train_dataset(pattern, batch_size):
    dataset = load_dataset(pattern, batch_size, num_repeat=None)
    return dataset.prefetch(1)