Python随机森林代码（导入数据，用于二分类和多分类问题）

最新推荐文章于 2024-07-15 15:30:12 发布

Cohen_ina

最新推荐文章于 2024-07-15 15:30:12 发布

阅读量6.7k

点赞数 8

文章标签： python 机器学习

本文链接：https://blog.csdn.net/weixin_45615071/article/details/120165167

版权

#!/user/bin/env python
# -*- coding:utf-8 -*-
#@Time    : 2021/9/7  13:32
#@Author  : Cohen
#@File    : RandomForest2.py
from typing import Any

import pandas as pd
import numpy as np
from sklearn.model_selection import train_test_split
from sklearn.ensemble import RandomForestClassifier
from matplotlib import pyplot as plt
from sklearn.metrics import plot_confusion_matrix

"""
@function: Reads training data from the specified file
@param1: The file name to be read by the function
@return1: The data set of the sample
@return2: The actual classification results of samples
@author: Cohen
@data: 2021/09/07
@warning: 使用时先把数据集填入到excel中，然后复制到文本文件中
"""
def loadDataSet(fileName,low):
    numFeat = len(open(fileName).readline