def saveLightsideGeneratedAnswers(essays, fname, author): with open(DATA_PATH + fname, "w") as csvfile: writer = csv.writer(csvfile) writer.writerow(["text", "author"]) for ess_id in sorted(essays): ess = essays[ess_id] writer.writerow([ess.getText(), author])
def saveLightsideTrainingAnswers(essays, fname, scoreType="domain1_score"): with open(DATA_PATH + fname, "w") as csvfile: writer = csv.writer(csvfile) writer.writerow(["label", "text"]) for ess_id in sorted(essays): ess = essays[ess_id] writer.writerow([ess.getScore(scoreType), ess.getText()])
def saveEssaysAsCsv(essays, fname="training.csv"): """ takes in essays, a dict from ids to Essay objects, and an output filename; writes those essays to a file """ with open(DATA_PATH + fname, "w") as csvfile: writer = csv.writer(csvfile) writer.writerow(header) for ess_id in sorted(essays): ess = essays[ess_id] writer.writerow( [ess.essay_id, ess.prompt_id, ess.getText()] + [ess.getScore(header[i]) for i in range(3, len(header))] )
def addTrainingAnswer(self, essay): data = {"corpus": lsrh.getCorpusURL(self.corpus_id), "text": essay.getText()} r = self.session.post(lsrh.getTrainingAnswerURL(), data=json.dumps(data)) self.trainingAnswerMap[essay.essay_id] = lsrh.getIdFromResponse(r)