isi-vista / adam

Abduction to Demonstrate an Articulate Machine

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

The 0th prediction is used for all objects if `--dir-num` is specified

boyleconnor opened this issue · comments

The else 0 in the line below seems to make it so the prediction for the 0th object of the curriculum will be used for every object in the whole curriculum if --dir-num is specified. Why?

predicted_label_ints[situation_num if args.dir_num is None else 0][i]] for i in range(args.top_k)

--dir-num makes it run on just a single curriculum situation directory, so we only have one prediction.