diff --git a/function.py b/function.py index 1fb84d4..98d9e56 100644 --- a/function.py +++ b/function.py @@ -100,6 +100,7 @@ def load_data(file_path): + if __name__ == "__main__": train_file = "datingTrainSet2.tsv" train_data, train_labels = load_data(train_file)