add genResult

finish some data exploration, add the genResult method
chdd · Aug 26, 2015 · 89c2ffe · 89c2ffe
1 parent 7957a79
commit 89c2ffe
Show file tree

Hide file tree

Showing 8 changed files with 3,593 additions and 4 deletions.
diff --git a/.gitignore b/.gitignore
@@ -6,3 +6,7 @@ data/weibo_predict_data.txt
 test.py
 
 *.pyc
+
+data/*
+
+result/*
diff --git a/.ipynb_checkpoints/OfflineTest-checkpoint.ipynb b/.ipynb_checkpoints/OfflineTest-checkpoint.ipynb
@@ -0,0 +1,9 @@
+{
+ "metadata": {
+  "name": "",
+  "signature": "sha256:480a61f83fafd9111c6a52450d3a81014ae1526eb4f01a61b956bf134e17ba02"
+ },
+ "nbformat": 3,
+ "nbformat_minor": 0,
+ "worksheets": []
+}
diff --git a/DataExploration.ipynb b/DataExploration.ipynb
diff --git a/OfflineTest.ipynb b/OfflineTest.ipynb
diff --git a/document/1202.0332v1.pdf b/document/1202.0332v1.pdf
diff --git a/document/Understanding_Retweeting_Behaviors_in_Social_Networks.pdf b/document/Understanding_Retweeting_Behaviors_in_Social_Networks.pdf
diff --git a/document/基于SVM的微博转发规模预测方法.pdf b/document/基于SVM的微博转发规模预测方法.pdf
diff --git a/weiboPredict/framework.py b/weiboPredict/framework.py
@@ -3,8 +3,11 @@
 # Author: Chen Li
 import pandas as pd
 import csv
+import re
+
 weibo_train_data = None
 weibo_predict_data = None
+
 def loadData():
 	global weibo_train_data 
 	weibo_train_data= pd.read_csv('data/weibo_train_data.txt',sep='\t', 
@@ -13,5 +16,17 @@ def loadData():
 	weibo_predict_data = pd.read_csv('data/weibo_predict_data.txt',sep='\t',
 		quoting=csv.QUOTE_NONE,names=['uid','mid','time','context'])
 
+def genResult(file, data):
+	data.to_csv('result/'+file+'.csv',sep=',',float_format='%d')
+	data.to_csv('result/'+file+'.txt',sep=',',float_format='%d',index=False,header=False)
+	f=open('result/'+file+'.txt','r')
+	context = f.read()
+	f.close()
+	context = re.sub(',(?=\w\w)','\t',context)
+	context = re.sub(',(?=\d,\d,\d)','\t',context)
+	f=open('result/test.txt','w')
+	f.write(context)
+	f.close()
+
 if __name__ == "__main__":
 	loadData()
-Original file line number
+Diff line change
@@ Expand Up / @@ -6,3 +6,7 @@ data/weibo_predict_data.txt @@
     test.py
     *.pyc
+    data/*
+    result/*