Aquila-Consortium/csiborgtools
0
0
Fork 0
mirror of https://github.com/Richard-Sti/csiborgtools_public.git synced 2025-06-30 19:41:12 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add batch size to submission

Browse source
This commit is contained in:
rstiskalek 2023-04-01 07:15:44 +01:00
parent 070b9b6c1b
commit 8d933e7ba9
2 changed files with 10 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

5
scripts/run_knn.py
View file

@ -42,6 +42,7 @@ parser.add_argument("--rmax", type=float)
parser.add_argument("--nneighbours", type=int)
parser.add_argument("--nsamples", type=int)
parser.add_argument("--neval", type=int)
parser.add_argument("--batch_size", type=int)
parser.add_argument("--seed", type=int, default=42)
args = parser.parse_args()
@ -77,8 +78,8 @@ def do_task(ic):
rs, cdf = knncdf(knn, nneighbours=args.nneighbours, Rmax=Rmax,
rmin=args.rmin, rmax=args.rmax, nsamples=args.nsamples,
neval=args.neval, random_state=args.seed,
verbose=False)
neval=args.neval, batch_size=args.batch_size,
random_state=args.seed, verbose=False)
out.update({"cdf_{}".format(i): cdf})
out.update({"rs": rs, "mass_threshold": mass_threshold})