2012-02-15 9 views

答えて

1

svmlight-loaderモジュールは、svmlightファイルをnumpy配列にロードできます。私は他の方向には何も存在しないと思うが、おそらくモジュールはその機能を拡張するための良い出発点になるだろう。

5

私はこの完全に最適化されていないスクリプトを少し前に書きましたが、たぶん役立つかもしれません!データとラベルは、2つの別々の配列でなければなりません。

def save_svmlight_data(data, labels, data_filename, data_folder = ''): 
    file = open(data_folder+data_filename,'w') 

    for i,x in enumerate(data): 
     indexes = x.nonzero()[0] 
     values = x[indexes] 

     label = '%i'%(labels[i]) 
     pairs = ['%i:%f'%(indexes[i]+1,values[i]) for i in xrange(len(indexes))] 

     sep_line = [label] 
     sep_line.extend(pairs) 
     sep_line.append('\n') 

     line = ' '.join(sep_line) 

     file.write(line) 
関連する問題