Esempio n. 1
0
def main():
	kernelFile = '/afs/cs.stanford.edu/u/rwitten/projects/multi_kernel_spl/data/allkernels_info.txt'
	trainFile = './train.newsmall_1_reducedy.txt' #TODO: Rafi, move this to your data directory so it doesn't clutter things up

	params = Params()
	spl_params = Params()
	spl_params.spl_mode = 0
	params.max_outer_iter = 1337 #TODO: make this user input
	loadKernelFile(kernelFile, params) 
	loadTrainFile(trainFile, params)

	w = ImagePsi.PsiObject(params)
	LSSVM.optimize(w, params, spl_params)
 
	return params
Esempio n. 2
0
def main():
	try:
		params = UserInput.getUserInput('train')	
		ExampleLoader.loadExamples(params)
		CommonApp.setExampleCosts(params)
		w = None
		if params.initialModelFile:
			w = CacheObj.loadObject(params.initialModelFile)
		else:
			w = CommonApp.PsiObject(params,False)

		globalSPLVars = SPLSelector.SPLVar()
		
		if params.splParams.splMode != 'CCCP':
			SPLSelector.setupSPL(params)
	
		w = LSSVM.optimize(w, globalSPLVars, params)
		CacheObj.cacheObject(params.modelFile,w)
		Performance.printStrongAndWeakTrainError(params, w)
	except Exception, e :
		import traceback
		traceback.print_exc(file=sys.stdout)