From ffe9db0f0032aa2762beaee22ce5b80477cedab9 Mon Sep 17 00:00:00 2001 From: maelstrom Date: Sat, 7 Dec 2024 22:07:44 -0500 Subject: [PATCH] typo --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 22e3608..d550750 100644 --- a/train.py +++ b/train.py @@ -20,7 +20,7 @@ def main(): spark = SparkSession.builder.appName("train").getOrCreate() pipe = SpectrogramPipe(spark) data = pipe.spectrogram_pipe(path, labels) - train_df, validation_df, test_df = data.randomsplit([0.8, 0.1, 0.1], + train_df, validation_df, test_df = data.randomSplit([0.8, 0.1, 0.1], seed=42) print(train_df.count()) print(validation_df.count())