def setUp(self):
self.u = User.objects.create_user(username='******', password='******')
self.job = Job.objects.create_active(
account=self.u.get_profile(),
gold_samples=[{'url': '10clouds.com', 'label': LABEL_YES}])
self.train_data = [
Sample(job=self.job, source_type='',
text='Mechanical squirrel screwdriver over car'),
Sample(job=self.job, source_type='',
text='Screwdriver fix mechanical bike bolts'),
Sample(job=self.job, source_type='',
text='Brown banana apple pinapple potato'),
Sample(job=self.job, source_type='',
text='apple pinapple potato'),
Sample(job=self.job, source_type='',
text='Hippo tree over lagoon'),
Sample(job=self.job, source_type='',
text='Green tan with true fox')
]
self.labels = [LABEL_YES, LABEL_YES, LABEL_NO, LABEL_NO, LABEL_NO, LABEL_NO]
self.classified = []
for idx, sample in enumerate(self.train_data):
self.classified.append(ClassifiedSample.objects.create(
job=self.job,
sample=sample,
label=self.labels[idx]
))
self.classifier247 = classifier_factory.create_classifier(
job_id=self.job.id,
)
