Spaces:
Runtime error
Runtime error
DRY
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ from src.utils import search_df
|
|
7 |
APP_PAGE_HEADER()
|
8 |
|
9 |
|
10 |
-
@st.cache
|
11 |
class LoadData:
|
12 |
train: pd.DataFrame = pd.read_csv("data/train.csv")
|
13 |
train = train.sample(frac=1).reset_index(drop=True) # shuffle data
|
@@ -40,7 +40,6 @@ class App:
|
|
40 |
def render_body(*args, **kwargs):
|
41 |
Helper().display_train_data()
|
42 |
|
43 |
-
st.header("Visualize Phrases as a Network Graph")
|
44 |
Helper().visualize()
|
45 |
|
46 |
def render_footer(self, *args, **kwargs):
|
@@ -50,18 +49,16 @@ class App:
|
|
50 |
class Helper(App):
|
51 |
|
52 |
def display_train_data(self):
|
53 |
-
data = self.data.
|
54 |
-
|
55 |
-
data = self._add_code_title(data)
|
56 |
|
57 |
st.write(f"> Train data `{data.shape[0]}` rows")
|
58 |
filter_ = st.text_input("search phrases", "")
|
59 |
if filter_:
|
60 |
-
data = search_df(
|
61 |
st.write(data)
|
62 |
|
63 |
def visualize(self, *args, **kwargs):
|
64 |
-
|
65 |
data = self.data.train_kg
|
66 |
|
67 |
# filter data for visualization
|
@@ -111,14 +108,6 @@ class Helper(App):
|
|
111 |
g.add_edge(n1, n2, title=label, show_edge_weights=True) # weight 42
|
112 |
return g
|
113 |
|
114 |
-
@st.cache
|
115 |
-
def _add_code_title(self, data):
|
116 |
-
titles = self.data.titles
|
117 |
-
data = data.merge(titles, left_on="context", right_on="code", how="left")
|
118 |
-
cols = data.columns.tolist()[:7]
|
119 |
-
cols.remove("code")
|
120 |
-
return data[cols]
|
121 |
-
|
122 |
|
123 |
if __name__ == "__main__":
|
124 |
app = App()
|
|
|
7 |
APP_PAGE_HEADER()
|
8 |
|
9 |
|
10 |
+
@st.cache(allow_output_mutation=True)
|
11 |
class LoadData:
|
12 |
train: pd.DataFrame = pd.read_csv("data/train.csv")
|
13 |
train = train.sample(frac=1).reset_index(drop=True) # shuffle data
|
|
|
40 |
def render_body(*args, **kwargs):
|
41 |
Helper().display_train_data()
|
42 |
|
|
|
43 |
Helper().visualize()
|
44 |
|
45 |
def render_footer(self, *args, **kwargs):
|
|
|
49 |
class Helper(App):
|
50 |
|
51 |
def display_train_data(self):
|
52 |
+
data = self.data.train_df
|
|
|
|
|
53 |
|
54 |
st.write(f"> Train data `{data.shape[0]}` rows")
|
55 |
filter_ = st.text_input("search phrases", "")
|
56 |
if filter_:
|
57 |
+
data = search_df(data, filter_)
|
58 |
st.write(data)
|
59 |
|
60 |
def visualize(self, *args, **kwargs):
|
61 |
+
st.header("Visualize Phrases as a Network Graph")
|
62 |
data = self.data.train_kg
|
63 |
|
64 |
# filter data for visualization
|
|
|
108 |
g.add_edge(n1, n2, title=label, show_edge_weights=True) # weight 42
|
109 |
return g
|
110 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
111 |
|
112 |
if __name__ == "__main__":
|
113 |
app = App()
|