typo

Nolte-Group · Dec 8, 2024 · ffe9db0 · ffe9db0
1 parent c6d8a75
commit ffe9db0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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())