model.py 文件源码

def GaussianNBLocalModel(localTrainFeature, localTestFeature, localTrainLabel, config):
    print 'train...'
    model = GaussianNB()
    model.fit(X = localTrainFeature.toarray(), y = localTrainLabel)
    print 'predict...'
    if config['prob'] == False:
        return model.predict(localTestFeature.toarray())
    else:
        return model.predict_log_proba(localTestFeature.toarray())

#-- Gaussian Navie Bayes online predict model frame