s464913/ium_464913
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Dockerfile

Browse Source
This commit is contained in:
Mateusz 2024-04-01 17:07:19 +02:00
parent e1b8cc77bb
commit 9a71a5ee0e
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
create-dataset.py
View File

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