o2w/talon
5
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Only use load compat if we can't load the classifier

Browse Source
This commit is contained in:
Yacine Filali
2017-05-24 13:29:59 -07:00
parent f5f7264077
commit 4acf05cf28
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
talon/signature/learning/classifier.py
View File

@@ -30,11 +30,14 @@ def train(classifier, train_data_filename, save_classifier_filename=None):
def load(saved_classifier_filename, train_data_filename): def load(saved_classifier_filename, train_data_filename):
"""Loads saved classifier. """ """Loads saved classifier. """
try:
return joblib.load(saved_classifier_filename)
except Exception:
import sys import sys
if sys.version_info > (3, 0): if sys.version_info > (3, 0):
return load_compat(saved_classifier_filename) return load_compat(saved_classifier_filename)
return joblib.load(saved_classifier_filename) raise
def load_compat(saved_classifier_filename): def load_compat(saved_classifier_filename):