s434765/ium_434765
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

data

Browse Source
This commit is contained in:
s434765 2021-04-24 21:18:57 +02:00
parent bf94b00b8f
commit 4c0566b5e9
7 changed files with 1648 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
Dockerfile
View File

@ -6,7 +6,10 @@ RUN apt install -y python3-pip
RUN apt install -y unzip
RUN pip3 install pandas
RUN pip3 install kaggle
RUN pip3 install torch
COPY ./get_data.sh ./
COPY ./main.py ./
COPY get_data.py ./
COPY ./neural_network.sh ./
COPY ./neural_network.py ./
RUN mkdir /.kaggle
RUN chmod -R 777 /.kaggle

544
data_test Normal file
View File

File diff suppressed because one or more lines are too long

1089
data_train Normal file
View File

File diff suppressed because one or more lines are too long

0
main.py → get_data.py
View File

2
get_data.sh
View File

@ -8,5 +8,5 @@ if kaggle datasets download -d sgonkaggle/youtube-trend-with-subscriber && unzip
head -n 544 "data_shuf" > "data_test"
head -n 1088 "data_shuf" | tail -n 544 > "data_dev"
head -n +1089 "data_shuf" > "data_train"
python3 main.py USvideos_modified.csv
python3 get_data.py USvideos_modified.csv
fi

8
neural_network.py Normal file
View File

@ -0,0 +1,8 @@
import torch
import pandas as pd
import torch.nn as nn
import torch.nn.functional as F
from torch.utils.data import Dataset, DataLoader
device = 'cpu'

2
neural_network.sh Executable file
View File

@ -0,0 +1,2 @@
#!/bin/bash
python3 neural_network.py