使用kaggle跑李宏毅机器学习作业

最新推荐文章于 2024-05-21 21:30:26 发布

loco_monkey

最新推荐文章于 2024-05-21 21:30:26 发布

阅读量1.3k

点赞数

分类专栏：李宏毅_机器学习_homework 文章标签：机器学习人工智能 python

本文链接：https://blog.csdn.net/loco_monkey/article/details/125742863

版权

李宏毅_机器学习_homework 专栏收录该内容

4 篇文章 22 订阅

订阅专栏

以hw4为例
注册kaggle账号需要tizi来获取验证码，之后登录不需要tizi
使用GPU需要填写手机号，也需要tizi进行验证码验证，每周有37小时的gpu使用时间，也可以使用cpu
在这里插入图片描述

1.首先读取训练数据

根据右侧目录结构
在这里插入图片描述
修改读取数据集的路径

def parse_args():
	"""arguments"""
	config = {
	#修改读取数据集的路径
		"data_dir": "/kaggle/input/ml2022spring-hw4/Dataset",
		"save_path": "model.ckpt",
		"batch_size": 32,
		"n_workers": 8,
		"valid_steps": 2000,#2000
		"warmup_steps": 1000, #1000
		"save_steps": 10000, #10000
		"total_steps": 70000, #70000
	}

2.读取训练集地址/保存模型地址/test数据集预测结果

def parse_args():
	"""arguments"""
	config = {
		"data_dir": "/kaggle/input/ml2022spring-hw4/Dataset",
		"model_path": "/kaggle/working/model.ckpt",
		"output_path": "/kaggle/working/output.csv",
	}

	return config