diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/app.py | 2 | ||||
| -rw-r--r-- | src/helpers/preprocessor.py | 1 | ||||
| -rw-r--r-- | src/windows/modeltrainer.py | 8 |
3 files changed, 10 insertions, 1 deletions
@@ -20,6 +20,8 @@ if DatabaseHandler().read().empty: if response: posts = fetch_data() # ask if user wants to save the data + root = tkinter.Tk() + root.withdraw() response = tkinter.messagebox.askokcancel('Save Data', 'Do you want to save the data for future use?', icon='warning') posts = pd.DataFrame(posts) posts = Preprocessor(posts).get_preprocessed_data() diff --git a/src/helpers/preprocessor.py b/src/helpers/preprocessor.py index 692e036..c2e138b 100644 --- a/src/helpers/preprocessor.py +++ b/src/helpers/preprocessor.py @@ -1,6 +1,5 @@ import pandas as pd - class Preprocessor: def __init__(self, dataframe): self.df = dataframe diff --git a/src/windows/modeltrainer.py b/src/windows/modeltrainer.py index 1d332fe..6664b57 100644 --- a/src/windows/modeltrainer.py +++ b/src/windows/modeltrainer.py @@ -11,6 +11,14 @@ from nltk.stem import SnowballStemmer from scipy.sparse import hstack from sklearn.feature_extraction.text import TfidfVectorizer from sklearn.preprocessing import LabelBinarizer +import ssl + +try: + _create_unverified_https_context = ssl._create_unverified_context +except AttributeError: + pass +else: + ssl._create_default_https_context = _create_unverified_https_context warnings.filterwarnings('ignore') nltk.download('stopwords') |