hylee719 commited on
Commit
631742f
·
1 Parent(s): b1a5ca2

minor fixes

Browse files
Files changed (1) hide show
  1. handler.py +12 -12
handler.py CHANGED
@@ -36,7 +36,10 @@ class Utterance:
36
  self.role = None
37
  self.word_count = self.get_num_words()
38
  self.timestamp = [starttime, endtime]
39
- self.unit_measure = None
 
 
 
40
  self.aggregate_unit_measure = endtime
41
  self.num_math_terms = None
42
  self.math_terms = None
@@ -138,11 +141,14 @@ class Transcript:
138
  utt.role = 'student'
139
  student_words += utt.get_num_words()
140
  student_utt_count += 1
141
- teacher_percentage = round(
142
- (teacher_words / (teacher_words + student_words)) * 100)
143
- student_percentage = 100 - teacher_percentage
144
- avg_teacher_length = teacher_words / teacher_utt_count
145
- avg_student_length = student_words / student_utt_count
 
 
 
146
  return {'teacher': teacher_percentage, 'student': student_percentage}, {'teacher': avg_teacher_length, 'student': avg_student_length}
147
 
148
  def get_word_cloud_dicts(self):
@@ -150,8 +156,6 @@ class Transcript:
150
  student_dict = {}
151
  uptake_teacher_dict = {}
152
  stop_words = stopwords.words('english')
153
- # stopwords = nltk.corpus.stopwords.word('english')
154
- # print("stopwords: ", stopwords)
155
  for utt in self.utterances:
156
  words = (utt.get_clean_text(remove_punct=True)).split(' ')
157
  for word in words:
@@ -411,10 +415,6 @@ class EndpointHandler():
411
  utterances = data.pop("inputs", data)
412
  params = data.pop("parameters", None)
413
 
414
- print("EXAMPLES")
415
- for utt in utterances[:3]:
416
- print("speaker %s: %s" % (utt["speaker"], utt["text"]))
417
-
418
  transcript = Transcript(filename=params.pop("filename", None))
419
  for utt in utterances:
420
  transcript.add_utterance(Utterance(**utt))
 
36
  self.role = None
37
  self.word_count = self.get_num_words()
38
  self.timestamp = [starttime, endtime]
39
+ if starttime is not None and endtime is not None:
40
+ self.unit_measure = endtime - starttime
41
+ else:
42
+ self.unit_measure = None
43
  self.aggregate_unit_measure = endtime
44
  self.num_math_terms = None
45
  self.math_terms = None
 
141
  utt.role = 'student'
142
  student_words += utt.get_num_words()
143
  student_utt_count += 1
144
+ if teacher_words + student_words > 0:
145
+ teacher_percentage = round(
146
+ (teacher_words / (teacher_words + student_words)) * 100)
147
+ student_percentage = 100 - teacher_percentage
148
+ else:
149
+ teacher_percentage = student_percentage = 0
150
+ avg_teacher_length = teacher_words / teacher_utt_count if teacher_utt_count > 0 else 0
151
+ avg_student_length = student_words / student_utt_count if student_utt_count > 0 else 0
152
  return {'teacher': teacher_percentage, 'student': student_percentage}, {'teacher': avg_teacher_length, 'student': avg_student_length}
153
 
154
  def get_word_cloud_dicts(self):
 
156
  student_dict = {}
157
  uptake_teacher_dict = {}
158
  stop_words = stopwords.words('english')
 
 
159
  for utt in self.utterances:
160
  words = (utt.get_clean_text(remove_punct=True)).split(' ')
161
  for word in words:
 
415
  utterances = data.pop("inputs", data)
416
  params = data.pop("parameters", None)
417
 
 
 
 
 
418
  transcript = Transcript(filename=params.pop("filename", None))
419
  for utt in utterances:
420
  transcript.add_utterance(Utterance(**utt))