def _list_outputs(self):
outputs = self.output_spec().get()
if isdefined(self.inputs.output_directory):
outputs['output_directory'] = Directory(exists=False, value=self.inputs.output_directory)
else:
outputs['output_directory'] = Directory(exists=False, value='accuracy_test')
return outputs
class AccuracyTesterInputSpec(CommandLineInputSpec):
mel_icas = InputMultiPath(Directory(exists=True), copyfile=False,
desc='Melodic output directories',
argstr='%s', position=3, mandatory=True)
trained_wts_file = File(desc='trained-weights file', argstr='%s', position=1, mandatory=True)
output_directory = Directory(desc='Path to folder in which to store the results of the accuracy test.', argstr='%s', position=2, mandatory=True)
class TrainingInputSpec(CommandLineInputSpec):
mel_icas = InputMultiPath(Directory(exists=True), copyfile=False,
desc='Melodic output directories',
argstr='%s', position=-1)
trained_wts_filestem = traits.Str(desc='trained-weights filestem, used for trained_wts_file and output directories', argstr='%s', position=1)
loo = traits.Bool(argstr='-l', desc='full leave-one-out test with classifier training', position=2)
class ClassifierInputSpec(CommandLineInputSpec):
mel_ica = Directory(exists=True, copyfile=False, desc='Melodic output directory or directories',
argstr='%s', position=1)
trained_wts_file = File(exists=True, desc='trained-weights file', argstr='%s', position=2, mandatory=True, copyfile=False)
thresh = traits.Int(argstr='%d', desc='Threshold for cleanup.', position=-1, mandatory=True)
artifacts_list_file = File(desc='Text file listing which ICs are artifacts; can be the output from classification or can be created manually')
class TrainingSetCreatorOutputSpec(TraitedSpec):
mel_icas_out = OutputMultiPath(Directory(exists=True),
copyfile=False,
desc='Hand labels for noise vs signal',
argstr='%s',
position=-1)
class TrainingSetCreatorInputSpec(BaseInterfaceInputSpec):
mel_icas_in = InputMultiPath(Directory(exists=True),
copyfile=False,
desc='Melodic output directories',
argstr='%s',
position=-1)
class AccuracyTesterOutputSpec(TraitedSpec):
output_directory = Directory(
desc=
'Path to folder in which to store the results of the accuracy test.',
argstr='%s',
position=1)
class FeatureExtractorOutputSpec(TraitedSpec):
mel_ica = Directory(exists=True,
copyfile=False,
desc='Melodic output directory or directories',
argstr='%s',
position=-1)