s424714/ium_424714
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

feat: update build_id argument

Browse Source
This commit is contained in:
Filip Patyk 2023-05-10 16:45:36 +02:00
parent b82a42d684
commit a34fe44254
2 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Jenkinsfile-eval
View File

@ -67,7 +67,7 @@ node {
{
sh 'mkdir -p ./.cache'
// sh ""
sh 'python ./src/main.py --test '
sh 'python ./src/main.py --test --build_id=${currentBuild.number} '
sh 'python ./src/plot.py '
sh "cp ./results/*.csv ${WORKSPACE}"
sh "cp ./results/*.png ${WORKSPACE}"

7
src/main.py
View File

@ -21,6 +21,7 @@ parser.add_argument("--test", action="store_true", default=False)
parser.add_argument("--model_path", type=str, default="results/model.pt")
parser.add_argument("--results_path", type=str, default="results/results.csv")
parser.add_argument("--data_acc_path", type=str, default="./results/acc.csv")
parser.add_argument("--build_id", type=int, default=0)
# HYPER PARAMETERS
parser.add_argument("--batch", "-b", type=int, default=2)
@ -83,4 +84,8 @@ if __name__ == "__main__":
batch_size=BATCH_SIZE,
)
utils.save_results(labels=test_data["label"], results=results, file_path=args.results_path)
utils.save_data_to_csv(args.data_acc_path, accuracy)
utils.save_data_to_csv(
file_path=args.data_acc_path,
build_id=args.build_id,
data=accuracy,
)