本文整理汇总了Python中Data.get_dataset方法的典型用法代码示例。如果您正苦于以下问题:Python Data.get_dataset方法的具体用法?Python Data.get_dataset怎么用?Python Data.get_dataset使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类Data
的用法示例。
在下文中一共展示了Data.get_dataset方法的2个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: classify
# 需要导入模块: import Data [as 别名]
# 或者: from Data import get_dataset [as 别名]
def classify():
file = open('result.txt', 'w')
X_train, X_test, y_train, y_test = Data.get_dataset(0.3)
clf = OneVsRestClassifier(svm.SVC())
clf.fit(X_train, y_train)
joblib.dump(clf, 'svm_model.pkl')
y_prediction = clf.predict(X_test)
acc = accuracy_score(y_test, y_prediction, normalize=False)
print(acc)
file.write(str(acc))
file.close()
示例2: classify
# 需要导入模块: import Data [as 别名]
# 或者: from Data import get_dataset [as 别名]
def classify():
file = open('result_gbc.txt', 'w')
X_train, X_test, y_train, y_test = Data.get_dataset(0.3)
clf = OneVsRestClassifier(GradientBoostingClassifier(n_estimators=100))
clf.fit(X_train, y_train)
joblib.dump(clf, 'gbc_model.pkl')
y_prediction = clf.predict(X_test)
acc = accuracy_score(y_test, y_prediction)
print(acc)
file.write(str(acc))
file.close()