train_task_classify is searching for .rec in wrong folder

[sipian]

When I am running train_task_classify.py

ncc/train_task_classifyapp.py

Lines 285 to 291 in 4e9eaeb

	seq_files = [os.path.join(folder, f) for f in listing if f[-4:] == '.rec']
	# training: Randomly pick programs
	assert len(seq_files) >= samples_per_class, "Cannot sample " + str(samples_per_class) + " from " + str(
	len(seq_files)) + " files found in " + folder
	X_train += resample(seq_files, replace=False, n_samples=samples_per_class, random_state=seed)
	y_train = np.concatenate([y_train, np.array([int(i)] * samples_per_class, dtype=np.int32)])

This assertion is failing (seq_files list is empty).
This is because the .rec files are generated in separate folders prefixed by seq.

ncc/task_utils.py

Lines 259 to 269 in 4e9eaeb

	def llvm_ir_to_trainable(folder_ir):
	####################################################################################################################
	# Setup
	assert len(folder_ir) > 0, "Please specify a folder containing the raw LLVM IR"
	assert os.path.exists(folder_ir), "Folder not found: " + folder_ir
	folder_seq = re.sub('ir', 'seq', folder_ir)
	if len(folder_seq) > 0:
	print('Preparing to write LLVM IR index sequences to', folder_seq)
	if not os.path.exists(folder_seq):
	os.makedirs(folder_seq)

But train_task_classify is searching for .rec files in the original IR folder ir_train.

[natalymr]

Hello, @sipian

I had the same problem and used a workaround described here, the second message: #4
i.e. just change a directory.

[tbennun]

Thanks for reporting @sipian and thank you for mentioning the workaround, @natalymr.

I'll try with a clean build and replace the lines as necessary.

[tbennun]

@sipian Did #9 fix the issue?

[tbennun]

Closing the issue due to old age. Please reopen if necessary!