输入 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 (将#修改为@）