From 1f42dd374ed4f3d6489001dcf97cca9d77a8a845 Mon Sep 17 00:00:00 2001 From: s464953 Date: Thu, 9 May 2024 00:13:31 +0200 Subject: [PATCH] added model training --- Dockerfile | 7 ++----- model_creator.py | 1 - 2 files changed, 2 insertions(+), 6 deletions(-) diff --git a/Dockerfile b/Dockerfile index af6025c..d489d41 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,17 +1,14 @@ FROM ubuntu:latest -ENV KAGGLE_USERNAME=gulczas -ENV KAGGLE_KEY=default_key - RUN apt-get update && \ apt-get install -y \ - python3 \ + python3 \ python3-pip \ wget \ unzip \ && rm -rf /var/lib/apt/lists/* -RUN pip3 install pandas scikit-learn requests kaggle numpy +RUN pip3 install pandas scikit-learn requests numpy WORKDIR /app diff --git a/model_creator.py b/model_creator.py index fab60b2..555f6a4 100644 --- a/model_creator.py +++ b/model_creator.py @@ -3,7 +3,6 @@ import os import numpy as np import shutil import sys -from kaggle.api.kaggle_api_extended import KaggleApi from sklearn.model_selection import train_test_split from sklearn.linear_model import LogisticRegression from sklearn.metrics import accuracy_score