Skip to content

Commit da0fa16

Browse files
authored
removes extra imports
1 parent 0a55ee8 commit da0fa16

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

ds11mltoolkit/plot.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
from plotly import subplots
88
from collections import defaultdict
99
import plotly.offline as py
10-
from wordcloud import STOPWORDS
10+
from wordcloud import STOPWORDS, WordCloud
1111
import plotly.express as px
1212
from sklearn.metrics import auc, roc_curve
1313

@@ -144,8 +144,7 @@ def sunburst(df, interior:str, exterior:str, col_num:str, title:str):
144144
return fig
145145

146146
def wordcloudviz(column):
147-
import matplotlib.pyplot as plt
148-
from wordcloud import WordCloud
147+
149148
"""
150149
Function to create a quick visualization of wordclouds in a given column of a dataframe called df.
151150
@@ -251,4 +250,4 @@ def plot_roc_curve(y_true, y_pred, pos_label=1, figsize=(8, 8)):
251250
plt.ylabel('True Positive Rate')
252251
plt.title('Receiver operating characteristic (ROC) curve')
253252
plt.legend(loc="lower right")
254-
plt.show()
253+
plt.show()

0 commit comments

Comments
 (0)