Add attention dataset

kundankr94700 · Feb 24, 2014 · 8c89618 · 8c89618
1 parent ff48445
commit 8c89618
Show file tree

Hide file tree

Showing 3 changed files with 75 additions and 0 deletions.
diff --git a/attention.csv b/attention.csv
@@ -0,0 +1,61 @@
+,subject,attention,solutions,score
+0,1,divided,1,2.0
+1,2,divided,1,3.0
+2,3,divided,1,3.0
+3,4,divided,1,5.0
+4,5,divided,1,4.0
+5,6,divided,1,5.0
+6,7,divided,1,5.0
+7,8,divided,1,5.0
+8,9,divided,1,2.0
+9,10,divided,1,6.0
+10,11,focused,1,6.0
+11,12,focused,1,8.0
+12,13,focused,1,6.0
+13,14,focused,1,8.0
+14,15,focused,1,8.0
+15,16,focused,1,6.0
+16,17,focused,1,7.0
+17,18,focused,1,7.0
+18,19,focused,1,5.0
+19,20,focused,1,6.0
+20,1,divided,2,4.0
+21,2,divided,2,4.0
+22,3,divided,2,5.0
+23,4,divided,2,7.0
+24,5,divided,2,5.0
+25,6,divided,2,5.0
+26,7,divided,2,4.5
+27,8,divided,2,7.0
+28,9,divided,2,3.0
+29,10,divided,2,5.0
+30,11,focused,2,5.0
+31,12,focused,2,9.0
+32,13,focused,2,5.0
+33,14,focused,2,8.0
+34,15,focused,2,8.0
+35,16,focused,2,8.0
+36,17,focused,2,7.0
+37,18,focused,2,8.0
+38,19,focused,2,6.0
+39,20,focused,2,6.0
+40,1,divided,3,7.0
+41,2,divided,3,5.0
+42,3,divided,3,6.0
+43,4,divided,3,5.0
+44,5,divided,3,8.0
+45,6,divided,3,6.0
+46,7,divided,3,6.0
+47,8,divided,3,8.0
+48,9,divided,3,7.0
+49,10,divided,3,6.0
+50,11,focused,3,6.0
+51,12,focused,3,8.0
+52,13,focused,3,9.0
+53,14,focused,3,7.0
+54,15,focused,3,7.0
+55,16,focused,3,7.0
+56,17,focused,3,6.0
+57,18,focused,3,6.0
+58,19,focused,3,6.0
+59,20,focused,3,5.0
diff --git a/process/attention.py b/process/attention.py
@@ -0,0 +1,13 @@
+import pandas as pd
+
+def main():
+
+    df = pd.read_csv("raw/attention.csv")
+    df = pd.melt(df, ["subidr", "attnr"], var_name="solutions", value_name="score")
+    df.solutions = df.solutions.str[-1].astype(int)
+    df.columns = ["subject", "attention", "solutions", "score"]
+    df.to_csv("attention.csv")
+
+
+if __name__ == "__main__":
+    main()
diff --git a/raw/attention.csv b/raw/attention.csv
@@ -0,0 +1 @@
+subidr,attnr,num1,num2,num31,divided,2,4,72,divided,3,4,53,divided,3,5,64,divided,5,7,55,divided,4,5,86,divided,5,5,67,divided,5,4.5,68,divided,5,7,89,divided,2,3,710,divided,6,5,611,focused,6,5,612,focused,8,9,813,focused,6,5,914,focused,8,8,715,focused,8,8,716,focused,6,8,717,focused,7,7,618,focused,7,8,619,focused,5,6,620,focused,6,6,5
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		subidr,attnr,num1,num2,num31,divided,2,4,72,divided,3,4,53,divided,3,5,64,divided,5,7,55,divided,4,5,86,divided,5,5,67,divided,5,4.5,68,divided,5,7,89,divided,2,3,710,divided,6,5,611,focused,6,5,612,focused,8,9,813,focused,6,5,914,focused,8,8,715,focused,8,8,716,focused,6,8,717,focused,7,7,618,focused,7,8,619,focused,5,6,620,focused,6,6,5
Expand Down