Initial commit

Author: nzc

.idea/dictionaries/niezhaochang.xml

@@ -0,0 +1,13 @@
+# -*- coding:utf-8 -*-
+
+from utils import data_preprocess
+from model import DeepFM
+import torch
+
+result_dict = data_preprocess.read_criteo_data('./data/tiny_train_input.csv', './data/category_emb.csv')
+test_dict = data_preprocess.read_criteo_data('./data/tiny_test_input.csv', './data/category_emb.csv')
+
+with torch.cuda.device(2):
+    deepfm = DeepFM.DeepFM(39,result_dict['feature_sizes'],verbose=True,use_cuda=True, weight_decay=0.0001,use_fm=False,use_ffm=True,use_deep=False).cuda()
+    deepfm.fit(result_dict['index'][0:2018], result_dict['value'][0:2018], result_dict['label'][0:2018],
+               test_dict['index'][0:2018], test_dict['value'][0:2018], test_dict['label'][0:2018],ealry_stopping=True,refit=True)