Light-GCN数据集

本文介绍了Light-GCN在Gowalla数据集上的使用情况,该数据集包含用户签到记录,每个记录包括用户ID、签到时间和地点ID等信息。为了保证数据质量,采用10核心设置,剔除了交互少于10次的用户。数据预处理阶段,只保留User-ID和Location-ID,对不足10个交互的用户进行剔除,并对ID进行重新编排,最终随机划分得到训练和测试数据集。
摘要由CSDN通过智能技术生成

1>关于Light-GCN所使用的数据集----Gowalla

数据内容包括:好友关系数据集和签到数据集(好友关系数据集未使用,以下只介绍签到数据集)

图 1

       在签到数据集中,每个用户的每一次签到记录作为一条记录,签到的具体属性分列表示。每条签到记录包括用户ID(User-ID)、签到时间(Check-in Time)、签到地点的维度(Latitude)、签到地点的经度(Longitude)、以及每个地区的经纬度所唯一对应的地点ID(Location-ID)。其中,用户ID一一对应每个用户,经纬度精确到小数点第六位。

        数据集包括了196586个用户通过签到方式在5977758个地点上共享了6442892条行为信息。该数据记录在图1中的 loc-gowalla_totalVheckins.txt.gz 中。

        因为Light-GCN所使用的数据集与NGCF的数据集相同,NGCFGowalla做了相应的处理。为了保证数据集的质量,他们使用了10核心设置,即保留至少有十次交互的用户和项目。

  • 7
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 9
    评论
要训练ST-GCN模型使用自己的数据集,你可以按照以下步骤进行操作: 1. 准备数据集:首先,你需要准备一个包含动作示例的数据集。这个数据集可以是你自己收集的,也可以是从其他来源获取的。确保数据集的每个示例都包含一个动作的骨骼姿势信息和相应的标签。 2. 提取骨骼姿势:使用AlphaPose或其他类似的工具从视频提取骨骼姿势。确保你的数据集包含了每个动作示例的骨骼姿势序列。 3. 标记动作帧:对于每个动作示例,手动标记每个动作帧的标签。这将帮助模型学习动作的时序信息。 4. 数据预处理:对数据集进行预处理,包括将骨骼姿势序列转换为适合ST-GCN模型输入的格式。你可以根据ST-GCN的要求进行相应的处理,比如调整帧数、归一化等。 5. 构建训练集和验证集:将数据集划分为训练集和验证集。训练集用于训练模型的参数,验证集用于评估模型的性能和调整超参数。 6. 训练ST-GCN模型:使用训练集对ST-GCN模型进行训练。你可以使用已经预训练的Tiny-YOLO oneclass-和SPPE模型作为基础模型,然后在此基础上进行微调。 7. 调整超参数:根据验证集的性能,调整模型的超参数,比如学习率、批大小等,以进一步提高模型的性能。 8. 评估模型:使用验证集对训练好的模型进行评估,计算模型在动作识别任务上的准确率或其他指标。 9. 应用模型:一旦模型训练完成并通过验证集的评估,你可以将该模型应用于新的未见过的数据上,进行动作识别任务。 请注意,在训练自己的数据集时,你需要确保数据集的质量和多样性,以获得更好的模型性能。此外,你还可以参考引用提供的文章链接,了解更多关于ST-GCN模型训练自己的数据集的详细步骤。
评论 9
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值