28 lines
786 B
Python
28 lines
786 B
Python
import os
|
|
from resource_fetch import ResourceFetch
|
|
from coffeehouse_dltc.main import DLTC
|
|
|
|
__all__ = ['SpamDetection']
|
|
|
|
|
|
class SpamDetection(object):
|
|
|
|
def __init__(self):
|
|
"""
|
|
Public Constructor
|
|
"""
|
|
self.dltc = DLTC()
|
|
self.rf = ResourceFetch()
|
|
self.model_directory = os.path.join(self.rf.fetch("Intellivoid", "CoffeeHouseData-Spam"), 'chatrooms_build')
|
|
self.dltc.load_model_cluster(self.model_directory)
|
|
|
|
def predict(self, text_input):
|
|
"""
|
|
Takes the user input and predicts if the input is either
|
|
spam or ham
|
|
|
|
:param text_input:
|
|
:return: Returns dictionary "ham", "spam" prediction values
|
|
"""
|
|
return self.dltc.predict_from_text(text_input)
|