added model training

This commit is contained in:
s464953 2024-05-09 00:13:31 +02:00
parent b20f390d38
commit 1f42dd374e
2 changed files with 2 additions and 6 deletions

View File

@ -1,8 +1,5 @@
FROM ubuntu:latest
ENV KAGGLE_USERNAME=gulczas
ENV KAGGLE_KEY=default_key
RUN apt-get update && \
apt-get install -y \
python3 \
@ -11,7 +8,7 @@ RUN apt-get update && \
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

View File

@ -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