如何解决输入 Scikit-learn 预测器的提示
BaseEnsemble
是 Scikit-learn 预测器的正确超类/接口吗?我是 Python 新手,所以也许我没有以正确的方式思考问题。如果这是 Java,这将是 public class RandomForestClassifier implements Classifiable
(或其他),而我的方法将是 public Classifiable train_from_db()
。
我有一个准备并返回 RandomForestClassifier
的函数:
from sklearn.ensemble import RandomForestClassifier
from sklearn.ensemble import BaseEnsemble
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
def train_from_db():
X = []
y = []
for sample_features,sample_class in query_local_database():
X.append(sample_features)
y.append(sample_class)
clf = RandomForestClassifier(random_state=0)
clf.fit(X,y)
return clf
我向该函数添加了类型提示,以使其更易于在我的程序中的其他地方使用。我开始于:
def train_from_db() -> RandomForestClassifier:
这有效。我概括为:
def train_from_db() -> BaseEnsemble:
我的程序仍在运行,但这是一个好的方法吗?
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。