Dockerfile

This commit is contained in:
Mateusz 2024-04-01 17:07:19 +02:00
parent e1b8cc77bb
commit 9a71a5ee0e

View File

@ -10,7 +10,6 @@ from sklearn.model_selection import train_test_split
def download_kaggle_dataset():
print(sys.argv)
os.environ["KAGGLE_USERNAME"] = "vskyper"
os.environ["KAGGLE_KEY"] = sys.argv[1]
kaggle = KaggleApi()
@ -118,7 +117,7 @@ def main():
)
X_train, X_test, y_train, y_test = split_whole_data(df)
save_whole_data(X_train, X_test, y_train, y_test)
save_whole_data(df, X_train, X_test, y_train, y_test)
if __name__ == "__main__":