当前位置：网站首页>使用kaggle跑李宏毅机器学习作业

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

2022-07-17 04:08:00 【loco_monkey】

以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