栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

sklearn:classification

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

sklearn:classification

classification_report的格式

引用:sklearn.metrics.classification_report模块使用与指标分析(生成混淆矩阵评价分类指标)

在使用autoclassfication后获取其中的precision recall f1-score 值

def auto_sklearn_classification(X_train, X_test, y_train, y_test):
    cls = autosklearn.classification.AutoSklearnClassifier(time_left_for_this_task=300, per_run_time_limit=90, ml_memory_limit=10000)
    cls.fit(X_train, y_train)
    predictions = cls.predict(X_test)

    report = []
    report_str = classification_report(y_test, predictions)
    for row in report_str.split("n"):
        parsed_row = [x for x in row.split("  ") if len(x) > 0]
        if len(parsed_row) > 0:
            report.append(parsed_row)
    
    # save accuracy, precision, recall, F1-Score to dictionary
    accuracy = accuracy_score(predictions,y_test)
    precision = float(report[-1][1])#最后一行第二列的值
    recall = float(report[-1][2])
    f1_score = float(report[-1][3])
    return accuracy, precision, recall, f1_score
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/688813.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号