Python复现An active learning-based SVM multi-class classification model

最新推荐文章于 2022-03-28 13:19:37 发布

DeniuHe

最新推荐文章于 2022-03-28 13:19:37 发布

阅读量187

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/DeniuHe/article/details/116722072

版权

本文介绍如何使用Python构建一个基于主动学习的SVM多类别分类模型。通过CBA（Coactive Bayesian Active Learning）方法，该模型能够在数据标注有限的情况下，有效地提升分类性能。

摘要由CSDN通过智能技术生成

CBA：

import xlwt
import numpy as np
import pandas as pd
from copy import deepcopy
from sklearn.metrics import accuracy_score, mean_absolute_error, f1_score, recall_score
from collections import OrderedDict
from sklearn.linear_model import LogisticRegression
from sklearn.model_selection import StratifiedKFold
from pathlib import Path
import matplotlib.pyplot as plt
from scipy.spatial.distance import pdist, squareform
from sklearn.svm import SVC

class CBA():
    def __init__(self, X_pool, y_pool, labeled, budget, X_test, y_test):
        self.X_pool = X_pool
        self.y_pool = y_pool.astype(np.int)
        self.X_test = X_test
        self.y_test = y_test.astype(np.int)
        self.labels = np.sort(np.unique(self.y_pool))
        self.nClass = len(self.labels)
        self.label