feat: update build_id argument
This commit is contained in:
parent
b82a42d684
commit
a34fe44254
@ -67,7 +67,7 @@ node {
|
|||||||
{
|
{
|
||||||
sh 'mkdir -p ./.cache'
|
sh 'mkdir -p ./.cache'
|
||||||
// sh ""
|
// sh ""
|
||||||
sh 'python ./src/main.py --test '
|
sh 'python ./src/main.py --test --build_id=${currentBuild.number} '
|
||||||
sh 'python ./src/plot.py '
|
sh 'python ./src/plot.py '
|
||||||
sh "cp ./results/*.csv ${WORKSPACE}"
|
sh "cp ./results/*.csv ${WORKSPACE}"
|
||||||
sh "cp ./results/*.png ${WORKSPACE}"
|
sh "cp ./results/*.png ${WORKSPACE}"
|
||||||
|
@ -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("--model_path", type=str, default="results/model.pt")
|
||||||
parser.add_argument("--results_path", type=str, default="results/results.csv")
|
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("--data_acc_path", type=str, default="./results/acc.csv")
|
||||||
|
parser.add_argument("--build_id", type=int, default=0)
|
||||||
|
|
||||||
# HYPER PARAMETERS
|
# HYPER PARAMETERS
|
||||||
parser.add_argument("--batch", "-b", type=int, default=2)
|
parser.add_argument("--batch", "-b", type=int, default=2)
|
||||||
@ -83,4 +84,8 @@ if __name__ == "__main__":
|
|||||||
batch_size=BATCH_SIZE,
|
batch_size=BATCH_SIZE,
|
||||||
)
|
)
|
||||||
utils.save_results(labels=test_data["label"], results=results, file_path=args.results_path)
|
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,
|
||||||
|
)
|
||||||
|
Loading…
Reference in New Issue
Block a user