コード例 #1
0
ファイル: learn_scoal.py プロジェクト: neerajg/sdap
def learn_scoal_wrapper(**argv):
    Z, W, rowAttr, colAttr, crossAttr, learner, params, \
    train_loss, test_loss, \
    num_cv, init_K, init_L, model_filename = read_options(**argv)
    
    Z = Z.tocsr()
    W = W.tocsr()
    param = params[0]
    I,J= sp.find(W)[:2]
    num_data = len(I)
    model = GeneralScoalModel()
    model.set_attributes(rowAttr, colAttr, crossAttr)
    #learn_scoal(model, Z, W, K, L, learner, param, train_loss, test_loss)
    model.save(model_filename)