quori_survey_analysis/main.py at master · CMU-RASL/quori_survey_analysis · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
from helper import read_dof_csv, check_matches, check_matches, create_batch
from helper import response_hist, analyze_demographics
import numpy as np


dof_csv_filename = 'torso_arm_dof.csv'
qualtrics_filename = 'Batch 1 Data/batch01_qualtrics.csv'
mturk_filename = 'Batch 1 Data/batch01_mturk.csv'
qualtrics_filename2 = 'Batch 2 Data/batch02_qualtrics.csv'
mturk_filename2 = 'Batch 2 Data/batch02_mturk.csv'

video_results_folder = 'video_results'
constraint_results_folder = 'constraint_results'
single_video_folder = 'single_video'
single_constraint_folder = 'single_constraint'

#Step 1: Check MTurk - Qualtrics Matches
user_arr = check_matches(qualtrics_filename, mturk_filename)
user_arr2 = check_matches(qualtrics_filename2, mturk_filename2)
user_arr.extend(user_arr2)

# analyze_demographics(user_arr)

#Step 2: Movement Array
movement_arr = read_dof_csv(dof_csv_filename)

#Step 3: Create Survey Results Object
batch = create_batch(user_arr, movement_arr)

#Compare Emotions
# batch.compare_emotions()

#Demographics
# batch.test_demo_group()

#Basic Group Comparison
# batch.check_balanced()
# batch.test_group()

batch.interaction_effect()

#Creating CSV of results
# batch.create_csv()

#Response histogram
# response_hist(batch)

#Plot Single Video
# batch.plot_single_video(single_video_folder)