Spaces:
Runtime error
Runtime error
Delete udfPreprocess/cleaning,py
Browse files- udfPreprocess/cleaning,py +0 -144
udfPreprocess/cleaning,py
DELETED
@@ -1,144 +0,0 @@
|
|
1 |
-
import pandas as pd
|
2 |
-
import numpy as np
|
3 |
-
import string
|
4 |
-
import nltk
|
5 |
-
import spacy
|
6 |
-
import en_core_web_sm
|
7 |
-
import re
|
8 |
-
import streamlit as st
|
9 |
-
|
10 |
-
from haystack.nodes import PreProcessor
|
11 |
-
|
12 |
-
'''basic cleaning - suitable for transformer models'''
|
13 |
-
def basic(s):
|
14 |
-
"""
|
15 |
-
:param s: string to be processed
|
16 |
-
:return: processed string: see comments in the source code for more info
|
17 |
-
"""
|
18 |
-
# Text Lowercase
|
19 |
-
#s = s.lower()
|
20 |
-
# Remove punctuation
|
21 |
-
#translator = str.maketrans(' ', ' ', string.punctuation)
|
22 |
-
#s = s.translate(translator)
|
23 |
-
# Remove URLs
|
24 |
-
s = re.sub(r'^https?:\/\/.*[\r\n]*', ' ', s, flags=re.MULTILINE)
|
25 |
-
s = re.sub(r"http\S+", " ", s)
|
26 |
-
# Remove new line characters
|
27 |
-
#s = re.sub('\n', ' ', s)
|
28 |
-
|
29 |
-
# Remove distracting single quotes
|
30 |
-
#s = re.sub("\'", " ", s)
|
31 |
-
# Remove all remaining numbers and non alphanumeric characters
|
32 |
-
#s = re.sub(r'\d+', ' ', s)
|
33 |
-
#s = re.sub(r'\W+', ' ', s)
|
34 |
-
|
35 |
-
# define custom words to replace:
|
36 |
-
#s = re.sub(r'strengthenedstakeholder', 'strengthened stakeholder', s)
|
37 |
-
|
38 |
-
return s.strip()
|
39 |
-
|
40 |
-
|
41 |
-
def preprocessingForSDG(document):
|
42 |
-
|
43 |
-
"""
|
44 |
-
takes in haystack document object and splits it into paragraphs and applies simple cleaning.
|
45 |
-
Returns cleaned list of haystack document objects. One paragraph per object. Also returns pandas df and
|
46 |
-
list that contains all text joined together.
|
47 |
-
"""
|
48 |
-
|
49 |
-
preprocessor = PreProcessor(
|
50 |
-
clean_empty_lines=True,
|
51 |
-
clean_whitespace=True,
|
52 |
-
clean_header_footer=True,
|
53 |
-
split_by="word",
|
54 |
-
split_length=100,
|
55 |
-
split_respect_sentence_boundary=True,
|
56 |
-
split_overlap=4
|
57 |
-
)
|
58 |
-
for i in document:
|
59 |
-
docs_processed = preprocessor.process([i])
|
60 |
-
for item in docs_processed:
|
61 |
-
item.content = basic(item.content)
|
62 |
-
|
63 |
-
st.write("your document has been splitted to", len(docs_processed), "paragraphs")
|
64 |
-
|
65 |
-
# create dataframe of text and list of all text
|
66 |
-
df = pd.DataFrame(docs_processed)
|
67 |
-
all_text = " ".join(df.content.to_list())
|
68 |
-
par_list = df.content.to_list()
|
69 |
-
|
70 |
-
return docs_processed, df, all_text, par_list
|
71 |
-
|
72 |
-
def preprocessing(document):
|
73 |
-
|
74 |
-
"""
|
75 |
-
takes in haystack document object and splits it into paragraphs and applies simple cleaning.
|
76 |
-
Returns cleaned list of haystack document objects. One paragraph per object. Also returns pandas df and
|
77 |
-
list that contains all text joined together.
|
78 |
-
"""
|
79 |
-
|
80 |
-
preprocessor = PreProcessor(
|
81 |
-
clean_empty_lines=True,
|
82 |
-
clean_whitespace=True,
|
83 |
-
clean_header_footer=True,
|
84 |
-
split_by="sentence",
|
85 |
-
split_length=3,
|
86 |
-
split_respect_sentence_boundary=False,
|
87 |
-
split_overlap=1
|
88 |
-
)
|
89 |
-
for i in document:
|
90 |
-
docs_processed = preprocessor.process([i])
|
91 |
-
for item in docs_processed:
|
92 |
-
item.content = basic(item.content)
|
93 |
-
|
94 |
-
st.write("your document has been splitted to", len(docs_processed), "paragraphs")
|
95 |
-
|
96 |
-
# create dataframe of text and list of all text
|
97 |
-
df = pd.DataFrame(docs_processed)
|
98 |
-
all_text = " ".join(df.content.to_list())
|
99 |
-
par_list = df.content.to_list()
|
100 |
-
|
101 |
-
return docs_processed, df, all_text, par_list
|
102 |
-
|
103 |
-
'''processing with spacy - suitable for models such as tf-idf, word2vec'''
|
104 |
-
def spacy_clean(alpha:str, use_nlp:bool = True) -> str:
|
105 |
-
|
106 |
-
"""
|
107 |
-
Clean and tokenise a string using Spacy. Keeps only alphabetic characters, removes stopwords and
|
108 |
-
filters out all but proper nouns, nounts, verbs and adjectives.
|
109 |
-
Parameters
|
110 |
-
----------
|
111 |
-
alpha : str
|
112 |
-
The input string.
|
113 |
-
use_nlp : bool, default False
|
114 |
-
Indicates whether Spacy needs to use NLP. Enable this when using this function on its own.
|
115 |
-
Should be set to False if used inside nlp.pipeline
|
116 |
-
Returns
|
117 |
-
-------
|
118 |
-
' '.join(beta) : a concatenated list of lemmatised tokens, i.e. a processed string
|
119 |
-
Notes
|
120 |
-
-----
|
121 |
-
Fails if alpha is an NA value. Performance decreases as len(alpha) gets large.
|
122 |
-
Use together with nlp.pipeline for batch processing.
|
123 |
-
"""
|
124 |
-
|
125 |
-
nlp = spacy.load("en_core_web_sm", disable=["parser", "ner", "textcat"])
|
126 |
-
|
127 |
-
if use_nlp:
|
128 |
-
|
129 |
-
alpha = nlp(alpha)
|
130 |
-
|
131 |
-
|
132 |
-
|
133 |
-
beta = []
|
134 |
-
|
135 |
-
for tok in alpha:
|
136 |
-
|
137 |
-
if all([tok.is_alpha, not tok.is_stop, tok.pos_ in ['PROPN', 'NOUN', 'VERB', 'ADJ']]):
|
138 |
-
|
139 |
-
beta.append(tok.lemma_)
|
140 |
-
|
141 |
-
|
142 |
-
text = ' '.join(beta)
|
143 |
-
text = text.lower()
|
144 |
-
return text
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|