def load_dataset()

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

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

def load_dataset(pattern, batch_size, num_repeat):
    dataset = tf.data.experimental.make_csv_dataset(
        file_pattern=pattern,
        batch_size=batch_size,
        column_names=CSV_COLS,
        column_defaults=DEFAULTS,
        num_epochs=num_repeat,
        shuffle_buffer_size=10000,
        header=True
    )
    return dataset.map(features_and_labels)