Package ml :: Package var :: Module extract_liblinear_weights
[hide private]
[frames] | no frames]

Source Code for Module ml.var.extract_liblinear_weights

 1  import cPickle as pickle 
 2  f=open('classifier.clsf') 
 3  classifier = pickle.load(f) 
 4  f.close() 
 5  weights_aligned = zip(classifier.__dict__['domain'],classifier.__dict__['weights'][0]) 
 6  weights_sorted = sorted(weights_aligned, key=lambda x: -abs(x[1])) 
 7  w=open('classifier.weights','w') 
 8   
 9  for feature, weight in weights_sorted: 
10      w.write('{}\t{}\n'.format(feature.name, weight)) 
11       
12  w.close() 
13