欢迎访问宙启技术站
智能推送

Python中的机器学习模型实践

发布时间:2023-12-25 07:55:47

在Python中,有很多机器学习模型可以用于各种任务,包括分类、回归、聚类等。下面是几个常见的机器学习模型的实践例子。

1.逻辑回归(Logistic Regression):

逻辑回归是一种常用的分类算法。它可以用于二分类和多分类问题。例如,我们可以使用逻辑回归模型来预测一个人是否患有某种疾病。下面是一个简单的逻辑回归实例:

import pandas as pd
from sklearn.linear_model import LogisticRegression

# 加载数据
data = pd.read_csv('data.csv')

# 提取特征和标签
X = data.drop('label', axis=1)
y = data['label']

# 初始化逻辑回归模型
model = LogisticRegression()

# 训练模型
model.fit(X, y)

# 进行预测
new_data = pd.read_csv('new_data.csv')
predictions = model.predict(new_data)

2.决策树(Decision Tree):

决策树是一种基于树结构的模型,用于分类和回归问题。它通过选择 的特征来划分数据,并生成一棵树。下面是一个决策树分类的示例:

import pandas as pd
from sklearn.tree import DecisionTreeClassifier
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score

# 加载数据
data = pd.read_csv('data.csv')

# 提取特征和标签
X = data.drop('label', axis=1)
y = data['label']

# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)

# 初始化决策树模型
model = DecisionTreeClassifier()

# 训练模型
model.fit(X_train, y_train)

# 进行预测
predictions = model.predict(X_test)

# 计算准确率
accuracy = accuracy_score(y_test, predictions)

3.支持向量机(Support Vector Machine, SVM):

支持向量机是一种用于二分类和多分类问题的监督学习模型。它通过在特征空间中找到一个最优超平面来完成分类任务。下面是一个使用支持向量机进行二分类的示例:

import pandas as pd
from sklearn.svm import SVC
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score

# 加载数据
data = pd.read_csv('data.csv')

# 提取特征和标签
X = data.drop('label', axis=1)
y = data['label']

# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)

# 初始化支持向量机模型
model = SVC()

# 训练模型
model.fit(X_train, y_train)

# 进行预测
predictions = model.predict(X_test)

# 计算准确率
accuracy = accuracy_score(y_test, predictions)

以上只是机器学习模型实践的几个例子,实际上在Python中还有很多其他的机器学习模型可以用于各种任务。选择适合的模型主要取决于问题的性质和数据的特征。