使用在 https://grouplens.org/datasets/movielens/ 上下载的MovieLens 数据集,并将使用以下文件:ratings.csv:包含用户对电影的评分,movies.csv:包含电影的详细信息 生成一个朴素贝叶斯方法的分类基于电影类型来预测用户可能喜欢的电影,并划分出训练集和测试集,输出f1值,auc值和roc曲线

时间: 2024-01-24 08:20:34 浏览: 150
7Z

YOLO 数据集:食材缺陷目标检测(10类)【包含划分好的数据集、类别class文件、数据可视化脚本】

好的,这是一个比较完整的数据分析任务,我们需要按照以下步骤进行: 1. 数据预处理 2. 特征工程 3. 模型训练与评估 我们先来看一下数据预处理。 # 1. 数据预处理 首先,我们需要将数据集读入,并做一些基本的预处理,例如数据清洗、缺失值处理等。 import pandas as pd import numpy as np # 读取数据集 ratings = pd.read_csv('ratings.csv') movies = pd.read_csv('movies.csv') # 将电影类型转换为二进制特征 genres = ["Action", "Adventure", "Animation", "Children", "Comedy", "Crime", "Documentary", "Drama", "Fantasy", "Film-Noir", "Horror", "Musical", "Mystery", "Romance", "Sci-Fi", "Thriller", "War", "Western"] for genre in genres: movies[genre] = movies['genres'].apply(lambda x: int(genre in x.split('|'))) movies = movies.drop('genres', axis=1) # 将电影ID转换为连续的整数 id2movie = dict(enumerate(list(movies['title']))) movie2id = {v: k for k, v in id2movie.items()} ratings['movieId'] = ratings['movieId'].apply(lambda x: movie2id[movies[movies['movieId'] == x]['title'].values[0]]) # 将用户ID转换为连续的整数 user2id = dict(enumerate(list(set(ratings['userId'])))) id2user = {v: k for k, v in user2id.items()} ratings['userId'] = ratings['userId'].apply(lambda x: user2id[x]) # 划分训练集和测试集 np.random.seed(123) msk = np.random.rand(len(ratings)) < 0.8 train = ratings[msk].reset_index(drop=True) test = ratings[~msk].reset_index(drop=True) # 查看数据集大小 print(train.shape, test.shape) # 输出示例数据 print(train.head()) print(test.head()) 以上代码将电影类型转换为二进制特征,并将电影ID和用户ID转换为连续的整数。然后我们将数据集随机拆分为训练集和测试集,比例为8:2。 接下来是特征工程。 # 2. 特征工程 我们将使用朴素贝叶斯方法来预测用户是否喜欢某个电影,因此需要将电影的类型作为特征。我们还可以使用其他一些特征,例如电影的平均评分和用户的平均评分。 # 计算每个电影的平均评分 movie_mean_rating = train.groupby('movieId').agg({'rating': 'mean'}).rename(columns={'rating': 'movie_mean_rating'}) # 计算每个用户的平均评分 user_mean_rating = train.groupby('userId').agg({'rating': 'mean'}).rename(columns={'rating': 'user_mean_rating'}) # 将平均评分添加到训练集和测试集中 train = pd.merge(train, movie_mean_rating, on='movieId', how='left') train = pd.merge(train, user_mean_rating, on='userId', how='left') test = pd.merge(test, movie_mean_rating, on='movieId', how='left') test = pd.merge(test, user_mean_rating, on='userId', how='left') # 将电影类型作为特征 X_train = pd.merge(train, movies.drop('title', axis=1), on='movieId', how='left').drop(['rating', 'timestamp'], axis=1) X_test = pd.merge(test, movies.drop('title', axis=1), on='movieId', how='left').drop(['rating', 'timestamp'], axis=1) # 将用户是否喜欢电影作为标签 y_train = (train['rating'] >= 4).astype(int) y_test = (test['rating'] >= 4).astype(int) 以上代码计算了每个电影和每个用户的平均评分,并将其作为特征添加到训练集和测试集中。然后将电影类型作为特征,并将用户是否喜欢电影作为标签。 接下来是模型训练与评估。 # 3. 模型训练与评估 我们将使用朴素贝叶斯方法来预测用户是否喜欢某个电影。 from sklearn.naive_bayes import GaussianNB from sklearn.metrics import f1_score, roc_auc_score, roc_curve # 训练模型 clf = GaussianNB() clf.fit(X_train, y_train) # 预测结果 y_pred = clf.predict(X_test) # 计算f1值和auc值 f1 = f1_score(y_test, y_pred) auc = roc_auc_score(y_test, y_pred) # 绘制ROC曲线 fpr, tpr, _ = roc_curve(y_test, y_pred) plt.plot(fpr, tpr) plt.xlabel('False Positive Rate') plt.ylabel('True Positive Rate') plt.title('ROC Curve') plt.show() print('f1 score:', f1) print('auc:', auc) 以上代码使用GaussianNB模型训练,并预测测试集结果。然后计算f1值和auc值,并绘制ROC曲线。 最后,我们可以输出f1值和auc值,并查看ROC曲线。 完整的代码如下:
阅读全文

相关推荐

解释这段代码:import os.path as osp import pandas as pd import torch from sentence_transformers import SentenceTransformer from torch_geometric.data import HeteroData, download_url, extract_zip from torch_geometric.transforms import RandomLinkSplit, ToUndirected url = 'https://files.grouplens.org/datasets/movielens/ml-latest-small.zip' root = osp.join(osp.dirname(osp.realpath(__file__)), '../../data/MovieLens') extract_zip(download_url(url, root), root) movie_path = osp.join(root, 'ml-latest-small', 'movies.csv') rating_path = osp.join(root, 'ml-latest-small', 'ratings.csv') def load_node_csv(path, index_col, encoders=None, **kwargs): df = pd.read_csv(path, index_col=index_col, **kwargs) mapping = {index: i for i, index in enumerate(df.index.unique())} x = None if encoders is not None: xs = [encoder(df[col]) for col, encoder in encoders.items()] x = torch.cat(xs, dim=-1) return x, mapping def load_edge_csv(path, src_index_col, src_mapping, dst_index_col, dst_mapping, encoders=None, **kwargs): df = pd.read_csv(path, **kwargs) src = [src_mapping[index] for index in df[src_index_col]] dst = [dst_mapping[index] for index in df[dst_index_col]] edge_index = torch.tensor([src, dst]) edge_attr = None if encoders is not None: edge_attrs = [encoder(df[col]) for col, encoder in encoders.items()] edge_attr = torch.cat(edge_attrs, dim=-1) return edge_index, edge_attr class SequenceEncoder(object): # The 'SequenceEncoder' encodes raw column strings into embeddings. def __init__(self, model_name='all-MiniLM-L6-v2', device=None): self.device = device self.model = SentenceTransformer(model_name, device=device) @torch.no_grad() def __call__(self, df): x = self.model.encode(df.values, show_progress_bar=True, convert_to_tensor=True, device=self.device) return x.cpu() class GenresEncoder(object)

最新推荐

recommend-type

python入门-30.寻找列表中只出现一次的数字-寻找单身狗.py

python入门-30.寻找列表中只出现一次的数字——寻找单身狗.py
recommend-type

布尔教育linux优化笔记

linux优化笔记,配套视频:https://www.bilibili.com/list/474327672?sid=4496133&spm_id_from=333.999.0.0&desc=1
recommend-type

知识付费系统-直播+讲师入驻+课程售卖+商城系统-v2.1.9版本搭建以及资源分享下载

知识付费系统-直播+讲师入驻+课程售卖+商城系统-v2.1.9版本搭建以及资源分享下载,CRMEB知识付费分销与直播营销系统是由西安众邦科技自主开发的一款在线教育平台,该系统不仅拥有独立的知识产权,还采用了先进的ThinkPhp5.0框架和Vue前端技术栈,集成了在线直播教学及课程分销等多种功能,旨在为用户提供全方位的学习体验,默认解压密码youyacaocom
recommend-type

美妆神域-JAVA-基于springBoot美妆神域设计与实现

美妆神域-JAVA-基于springBoot美妆神域设计与实现
recommend-type

原生js制作Google粘土logo动画涂鸦代码.zip

原生js制作Google粘土logo动画涂鸦代码.zip
recommend-type

火炬连体网络在MNIST的2D嵌入实现示例

资源摘要信息:"Siamese网络是一种特殊的神经网络,主要用于度量学习任务中,例如人脸验证、签名识别或任何需要判断两个输入是否相似的场景。本资源中的实现例子是在MNIST数据集上训练的,MNIST是一个包含了手写数字的大型数据集,广泛用于训练各种图像处理系统。在这个例子中,Siamese网络被用来将手写数字图像嵌入到2D空间中,同时保留它们之间的相似性信息。通过这个过程,数字图像能够被映射到一个欧几里得空间,其中相似的图像在空间上彼此接近,不相似的图像则相对远离。 具体到技术层面,Siamese网络由两个相同的子网络构成,这两个子网络共享权重并且并行处理两个不同的输入。在本例中,这两个子网络可能被设计为卷积神经网络(CNN),因为CNN在图像识别任务中表现出色。网络的输入是成对的手写数字图像,输出是一个相似性分数或者距离度量,表明这两个图像是否属于同一类别。 为了训练Siamese网络,需要定义一个损失函数来指导网络学习如何区分相似与不相似的输入对。常见的损失函数包括对比损失(Contrastive Loss)和三元组损失(Triplet Loss)。对比损失函数关注于同一类别的图像对(正样本对)以及不同类别的图像对(负样本对),鼓励网络减小正样本对的距离同时增加负样本对的距离。 在Lua语言环境中,Siamese网络的实现可以通过Lua的深度学习库,如Torch/LuaTorch,来构建。Torch/LuaTorch是一个强大的科学计算框架,它支持GPU加速,广泛应用于机器学习和深度学习领域。通过这个框架,开发者可以使用Lua语言定义模型结构、配置训练过程、执行前向和反向传播算法等。 资源的文件名称列表中的“siamese_network-master”暗示了一个主分支,它可能包含模型定义、训练脚本、测试脚本等。这个主分支中的代码结构可能包括以下部分: 1. 数据加载器(data_loader): 负责加载MNIST数据集并将图像对输入到网络中。 2. 模型定义(model.lua): 定义Siamese网络的结构,包括两个并行的子网络以及最后的相似性度量层。 3. 训练脚本(train.lua): 包含模型训练的过程,如前向传播、损失计算、反向传播和参数更新。 4. 测试脚本(test.lua): 用于评估训练好的模型在验证集或者测试集上的性能。 5. 配置文件(config.lua): 包含了网络结构和训练过程的超参数设置,如学习率、批量大小等。 Siamese网络在实际应用中可以广泛用于各种需要比较两个输入相似性的场合,例如医学图像分析、安全验证系统等。通过本资源中的示例,开发者可以深入理解Siamese网络的工作原理,并在自己的项目中实现类似的网络结构来解决实际问题。"
recommend-type

管理建模和仿真的文件

管理Boualem Benatallah引用此版本:布阿利姆·贝纳塔拉。管理建模和仿真。约瑟夫-傅立叶大学-格勒诺布尔第一大学,1996年。法语。NNT:电话:00345357HAL ID:电话:00345357https://theses.hal.science/tel-003453572008年12月9日提交HAL是一个多学科的开放存取档案馆,用于存放和传播科学研究论文,无论它们是否被公开。论文可以来自法国或国外的教学和研究机构,也可以来自公共或私人研究中心。L’archive ouverte pluridisciplinaire
recommend-type

L2正则化的终极指南:从入门到精通,揭秘机器学习中的性能优化技巧

![L2正则化的终极指南:从入门到精通,揭秘机器学习中的性能优化技巧](https://img-blog.csdnimg.cn/20191008175634343.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MTYxMTA0NQ==,size_16,color_FFFFFF,t_70) # 1. L2正则化基础概念 在机器学习和统计建模中,L2正则化是一个广泛应用的技巧,用于改进模型的泛化能力。正则化是解决过拟
recommend-type

如何构建一个符合GB/T19716和ISO/IEC13335标准的信息安全事件管理框架,并确保业务连续性规划的有效性?

构建一个符合GB/T19716和ISO/IEC13335标准的信息安全事件管理框架,需要遵循一系列步骤来确保信息系统的安全性和业务连续性规划的有效性。首先,组织需要明确信息安全事件的定义,理解信息安全事态和信息安全事件的区别,并建立事件分类和分级机制。 参考资源链接:[信息安全事件管理:策略与响应指南](https://wenku.csdn.net/doc/5f6b2umknn?spm=1055.2569.3001.10343) 依照GB/T19716标准,组织应制定信息安全事件管理策略,明确组织内各个层级的角色与职责。此外,需要设置信息安全事件响应组(ISIRT),并为其配备必要的资源、
recommend-type

Angular插件增强Application Insights JavaScript SDK功能

资源摘要信息:"Microsoft Application Insights JavaScript SDK-Angular插件" 知识点详细说明: 1. 插件用途与功能: Microsoft Application Insights JavaScript SDK-Angular插件主要用途在于增强Application Insights的Javascript SDK在Angular应用程序中的功能性。通过使用该插件,开发者可以轻松地在Angular项目中实现对特定事件的监控和数据收集,其中包括: - 跟踪路由器更改:插件能够检测和报告Angular路由的变化事件,有助于开发者理解用户如何与应用程序的导航功能互动。 - 跟踪未捕获的异常:该插件可以捕获并记录所有在Angular应用中未被捕获的异常,从而帮助开发团队快速定位和解决生产环境中的问题。 2. 兼容性问题: 在使用Angular插件时,必须注意其与es3不兼容的限制。es3(ECMAScript 3)是一种较旧的JavaScript标准,已广泛被es5及更新的标准所替代。因此,当开发Angular应用时,需要确保项目使用的是兼容现代JavaScript标准的构建配置。 3. 安装与入门: 要开始使用Application Insights Angular插件,开发者需要遵循几个简单的步骤: - 首先,通过npm(Node.js的包管理器)安装Application Insights Angular插件包。具体命令为:npm install @microsoft/applicationinsights-angularplugin-js。 - 接下来,开发者需要在Angular应用的适当组件或服务中设置Application Insights实例。这一过程涉及到了导入相关的类和方法,并根据Application Insights的官方文档进行配置。 4. 基本用法示例: 文档中提到的“基本用法”部分给出的示例代码展示了如何在Angular应用中设置Application Insights实例。示例中首先通过import语句引入了Angular框架的Component装饰器以及Application Insights的类。然后,通过Component装饰器定义了一个Angular组件,这个组件是应用的一个基本单元,负责处理视图和用户交互。在组件类中,开发者可以设置Application Insights的实例,并将插件添加到实例中,从而启用特定的功能。 5. TypeScript标签的含义: TypeScript是JavaScript的一个超集,它添加了类型系统和一些其他特性,以帮助开发更大型的JavaScript应用。使用TypeScript可以提高代码的可读性和可维护性,并且可以利用TypeScript提供的强类型特性来在编译阶段就发现潜在的错误。文档中提到的标签"TypeScript"强调了该插件及其示例代码是用TypeScript编写的,因此在实际应用中也需要以TypeScript来开发和维护。 6. 压缩包子文件的文件名称列表: 在实际的项目部署中,可能会用到压缩包子文件(通常是一些JavaScript库的压缩和打包后的文件)。在本例中,"applicationinsights-angularplugin-js-main"很可能是该插件主要的入口文件或者压缩包文件的名称。在开发过程中,开发者需要确保引用了正确的文件,以便将插件的功能正确地集成到项目中。 总结而言,Application Insights Angular插件是为了加强在Angular应用中使用Application Insights Javascript SDK的能力,帮助开发者更好地监控和分析应用的运行情况。通过使用该插件,可以跟踪路由器更改和未捕获异常等关键信息。安装与配置过程简单明了,但是需要注意兼容性问题以及正确引用文件,以确保插件能够顺利工作。