prashant
commited on
Commit
•
8ee6037
1
Parent(s):
98746bf
sorting sdg results
Browse files- utils/sdg_classifier.py +2 -0
utils/sdg_classifier.py
CHANGED
|
@@ -96,10 +96,12 @@ def sdg_classification(haystackdoc:List[Document])->Tuple[DataFrame,Series]:
|
|
| 96 |
x = x.rename('count')
|
| 97 |
x = x.rename_axis('SDG').reset_index()
|
| 98 |
x["SDG"] = pd.to_numeric(x["SDG"])
|
|
|
|
| 99 |
x['SDG_name'] = x['SDG'].apply(lambda x: _lab_dict[x])
|
| 100 |
x['SDG'] = x['SDG'].apply(lambda x: "SDG "+str(x))
|
| 101 |
df= df.drop(['Relevancy'], axis = 1)
|
| 102 |
df['SDG'] = pd.to_numeric(df['SDG'])
|
|
|
|
| 103 |
|
| 104 |
|
| 105 |
return df, x
|
|
|
|
| 96 |
x = x.rename('count')
|
| 97 |
x = x.rename_axis('SDG').reset_index()
|
| 98 |
x["SDG"] = pd.to_numeric(x["SDG"])
|
| 99 |
+
x = x.sort_values(by=['SDG'])
|
| 100 |
x['SDG_name'] = x['SDG'].apply(lambda x: _lab_dict[x])
|
| 101 |
x['SDG'] = x['SDG'].apply(lambda x: "SDG "+str(x))
|
| 102 |
df= df.drop(['Relevancy'], axis = 1)
|
| 103 |
df['SDG'] = pd.to_numeric(df['SDG'])
|
| 104 |
+
df = df.sort_values('SDG')
|
| 105 |
|
| 106 |
|
| 107 |
return df, x
|