def setOptions(optdict, train_or_test):
assert('--dataFile' in optdict)
dataFile = optdict['--dataFile']
assert(len(dataFile) > 0)
assert('--modelFile' in optdict)
modelFile = optdict['--modelFile']
assert(len(modelFile) > 0)
params = Params()
params.trainOrTest = train_or_test
params.splParams = Params()
params.epsilon = .01
params.C = 1.0
params.splParams.splMode = 'CCCP'
params.seed = 0
params.maxOuterIters = 10
params.minOuterIters = 5
params.estimatedNumConstraints = 100
params.syntheticParams = None
params.supervised = False
params.numYLabels = 20
params.maxPsiGap = 0.00001
params.splParams.splInitFraction = 0.5
params.splParams.splIncrement = 0.1
params.splParams.splInitIters = 1
params.splParams.splOuterIters = 1
params.splParams.splInnerIters = 1
params.babyData = 0
params.balanceClasses = 0
params.UCCCP = int(optdict['--UCCCP'])
if '--splMode' in optdict:
params.splParams.splMode = optdict['--splMode']
params.latentVariableFile = optdict['--latentVariableFile'] if '--latentVariableFile' in optdict else None
assert(params.latentVariableFile)
if params.splParams.splMode != 'CCCP':
assert('--splControl' in optdict)
if '--splControl' in optdict:
params.splParams.splControl = int(optdict['--splControl'])
if '--splInitIters' in optdict:
params.splParams.splInitIters = int(optdict['--splInitIters'])
if '--balanceClasses' in optdict:
params.balanceClasses = int(optdict['--balanceClasses'])
if '--babyData' in optdict:
params.babyData = int(optdict['--babyData'])
if params.babyData == 1:
params.numYLabels = 7
kernelFile = '/afs/cs.stanford.edu/u/rwitten/projects/multi_kernel_spl/data/allkernels_info.txt'
if '--maxPsiGap' in optdict:
params.maxPsiGap = float(optdict['--maxPsiGap'])
params.initialModelFile = None
if '--initialModelFile' in optdict:
params.initialModelFile = optdict['--initialModelFile']
if '--maxTimeIdle' in optdict:
params.maxTimeIdle = int(optdict['--maxTimeIdle'])
if '--supervised' in optdict:
params.supervised = optdict['--supervised']
if '--kernelFile' in optdict:
kernelFile = optdict['--kernelFile']
if '--C' in optdict:
params.C = float(optdict['--C'])
if '--epsilon' in optdict:
params.epsilon = float(optdict['--epsilon'])
if '--seed' in optdict:
params.seed = int(optdict['--seed'])
random.seed(params.seed)
numpy.random.seed(params.seed)
if '--maxOuterIters' in optdict:
params.maxOuterIters = int(optdict['--maxOuterIters'])
if '--numYLabels' in optdict:
params.numYLabels = int(optdict['--numYLabels'])
if '--synthetic' in optdict and int(optdict['--synthetic']):
params.syntheticParams = Params()
params.syntheticParams.numLatents = 10
params.syntheticParams.strength = 10.0
params.numYLabels = 5
params.maxPsiGap = 0.00001
params.numExamples = 50
params.totalLength = params.numYLabels + 1
params.lengthW = params.numYLabels * params.totalLength
if '--syntheticNumLatents' in optdict:
params.syntheticParams.numLatents = int(optdict['--syntheticNumLatents'])
if '--syntheticNumExamples' in optdict:
params.numExamples = int(optdict['--syntheticNumExamples'])
if '--syntheticStrength' in optdict:
params.syntheticParams.strength = float(optdict['--syntheticStrength'])
assert('--scratchFile' in optdict)
params.scratchFile = optdict['--scratchFile']
logFile = '%s.log' %params.scratchFile
try:
os.remove(logFile)
except OSError, e: # no such file
pass
