Пример #1
0
def main(ftrain, fdev=None, fmodel='model/model.pickle.gz'):
    # Load data
    print 'Loading training data ...'
    data = load(gzip.open(ftrain))
    M, labels = data['data'], data['labels']
    # Load dev data
    if fdev is not None:
        print 'Loading dev data ...'
        devdata = load(gzip.open(fdev))
        devM, devlabels = devdata['data'], devdata['labels']
    else:
        devM, devlabels = None, None
    # Training with specified parameters
    print 'Training ...'
    clf = Classifier()
    clf.train(M, labels, devM, devlabels)
    clf.savemodel(fmodel)
Пример #2
0
def main(ftrain, fdev=None, fmodel='model/model.pickle.gz'):
    # Load data
    print 'Loading training data ...'
    data = load(gzip.open(ftrain))
    M, labels = data['data'], data['labels']
    # Load dev data
    if fdev is not None:
        print 'Loading dev data ...'
        devdata = load(gzip.open(fdev))
        devM, devlabels = devdata['data'], devdata['labels']
    else:
        devM, devlabels = None, None
    # Training with specified parameters
    print 'Training ...'
    clf = Classifier()
    clf.train(M, labels, devM, devlabels)
    clf.savemodel(fmodel)