def test_GaussianNB(*data):
'''
Test Gaussian NB
:param data: train_data, test_data, train_value, test_value
:return: None
'''
X_train,X_test,y_train,y_test=data
cls=naive_bayes.GaussianNB()
cls.fit(X_train,y_train)
print('Training Score: {0}' .format( cls.score(X_train,y_train)))
print('Testing Score: {0}' .format( cls.score(X_test, y_test)))
评论列表
文章目录