def __init__(self, opt):
self.file = os.path.join(os.path.dirname(os.path.abspath(__file__)),
opt.file)
process = Process(self.file, opt)
self.inputs, self.labels, _ = process.get_data()
# feature length
self.p = process.feature_length
self.k = opt.k # 参数v 的维度
self.lr = opt.lr # 学习率
self.reg_l1 = opt.reg_l1
self.reg_l2 = opt.reg_l2
self.data_length = process.data_length # 数据总量
self.opt = opt
self.train_log = os.path.join(
os.path.dirname(os.path.abspath(__file__)), opt.train_log)
self.save_path = os.path.join(
os.path.dirname(os.path.abspath(__file__)), opt.save_path)
save_parent_path = os.path.split(self.save_path)[0]
if not os.path.exists(self.train_log):
os.makedirs(self.train_log)
if not os.path.exists(save_parent_path):
os.makedirs(save_parent_path)
def __init__(self, opt):
self.file = os.path.join(os.path.dirname(os.path.abspath(__file__)),
opt.file)
process = Process(self.file, opt)
self.inputs, self.labels, _ = process.get_data()
self.p = process.feature_length
if opt.if_all_feature:
self.feature2field = process.feature2field # ind 对应的域
self.p1 = self.p
else:
#就取前几维特征做测试
self.p1 = opt.feature_num
self.feature2field = {}
feature_ind = 0
for key, value in process.feature2field.items():
if feature_ind > self.p1:
break
self.feature2field[key] = value
feature_ind += 1
self.k = opt.k # 参数v 的维度
self.lr = opt.lr # 学习率
self.reg_l1 = opt.reg_l1
self.reg_l2 = opt.reg_l2
self.data_length = process.data_length # 数据总量
self.f = len(
process.fields_dict.keys()) - 2 #除去all_count feature2field
self.opt = opt
self.train_log = os.path.join(
os.path.dirname(os.path.abspath(__file__)), opt.train_log)
self.save_path = os.path.join(
os.path.dirname(os.path.abspath(__file__)), opt.save_path)
save_parent_path = os.path.split(self.save_path)[0]
if not os.path.exists(self.train_log):
os.makedirs(self.train_log)
if not os.path.exists(save_parent_path):
os.makedirs(save_parent_path)