fix little bugs

diary-training.py

@@ -41,7 +41,11 @@ exposure = exposure.loc[exposure["device_id"].isin(click_device_id)]
 # 打标签
 click["y"] = 1
 exposure["y"] = 0
-print("成功获取正负样本")
+
+print("正样本个数")
+print(click.shape[0])
+print("负样本个数")
+print(exposure.shape[0])
 
 # 合并点击表和曝光表
 data = click.append(exposure)
@@ -122,7 +126,8 @@ class FFMFormatPandas:
 ffm_train = FFMFormatPandas()
 data = ffm_train.fit_transform(data, y='y')
 print("done transform ffm")
-
+data.to_csv("/home/zhangyanzhao/data.csv",index=False)
+data = pd.read_csv("/home/zhangyanzhao/data.csv",header=None)
 n = np.rint(data.shape[0]/8)
 m = np.rint(data.shape[0]*(3/8))
 # 1/8的数据集用来做测试集