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,17 +1,14 @@
FROM ubuntu:latest FROM ubuntu:latest
ENV KAGGLE_USERNAME=gulczas
ENV KAGGLE_KEY=default_key
RUN apt-get update && \ RUN apt-get update && \
apt-get install -y \ apt-get install -y \
python3 \ python3 \
python3-pip \ python3-pip \
wget \ wget \
unzip \ unzip \
&& rm -rf /var/lib/apt/lists/* && 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 WORKDIR /app

View File

@ -3,7 +3,6 @@ import os
import numpy as np import numpy as np
import shutil import shutil
import sys import sys
from kaggle.api.kaggle_api_extended import KaggleApi
from sklearn.model_selection import train_test_split from sklearn.model_selection import train_test_split
from sklearn.linear_model import LogisticRegression from sklearn.linear_model import LogisticRegression
from sklearn.metrics import accuracy_score from sklearn.metrics import accuracy_score