Minor correction
This commit is contained in:
parent
c39f934912
commit
7c40272f24
|
@ -13,7 +13,7 @@ class SpamDetection(object):
|
|||
"""
|
||||
self.dltc = DLTC()
|
||||
self.rf = ResourceFetch()
|
||||
self.model_directory = os.path.join(self.rf.fetch("Intellivoid", "CoffeeHouseData-Spam"), 'spam_ham_build')
|
||||
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):
|
||||
|
|
Loading…
Reference in New Issue