This commit is contained in:
piotrwrzodak 2023-05-11 22:58:08 +02:00
parent cc25dfd3b2
commit e75c417504
3 changed files with 12 additions and 10 deletions

View File

@ -4,3 +4,10 @@ RUN apt-get update && \
apt-get install -y python3 python3-pip
RUN pip3 install pandas numpy
WORKDIR /app
COPY create-dataset.py /app
COPY data/barcelona_weekends.csv /app
CMD ["python3", "create-dataset.py"]

View File

@ -44,12 +44,7 @@ pipeline {
dockerfile true
}
steps {
sh 'python3 create-dataset.py'
}
}
stage('Archive') {
steps {
archiveArtifacts artifacts: 'data/barcelona_weekends.train.csv, data/barcelona_weekends.dev.csv, data/barcelona_weekends.test.csv', fingerprint: true
archiveArtifacts artifacts: 'barcelona_weekends.train.csv, barcelona_weekends.dev.csv, barcelona_weekends.test.csv', fingerprint: true
}
}
}

View File

@ -5,16 +5,16 @@ import numpy as np
cutoff = 10
data = pd.read_csv('./data/barcelona_weekends.csv')
data = pd.read_csv('./barcelona_weekends.csv')
data = data.sample(cutoff)
data = data.iloc[:, 1:]
train_set, dev_set, test_set = np.split(data.sample(frac=1, random_state=42),
[int(.6 * len(data)), int(.8 * len(data))])
train_set.to_csv('data/barcelona_weekends.train.csv', index=False)
dev_set.to_csv('data/barcelona_weekends.dev.csv', index=False)
test_set.to_csv('data/barcelona_weekends.test.csv', index=False)
train_set.to_csv('barcelona_weekends.train.csv', index=False)
dev_set.to_csv('barcelona_weekends.dev.csv', index=False)
test_set.to_csv('barcelona_weekends.test.csv', index=False)
check = pd.read_csv('./train.csv')