update

7828c7b2 · rico.liu · b0ef852d · 7828c7b2
Commit 7828c7b2 authored Apr 01, 2021 by rico.liu
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

public_function.py public/SKUBasicInfo/public_function.py +2 -2

No files found.
--- a/public/SKUBasicInfo/public_function.py
+++ b/public/SKUBasicInfo/public_function.py
@@ -1817,8 +1817,8 @@ def predict_category(name_list):
    print(os.getcwd())
    os.chdir("./public/SKUBasicInfo/text_moudle_LXWL")

-    tf.reset_default_graph()
-    #tf.compat.v1.reset_default_graph()
+    #tf.reset_default_graph()
+    tf.compat.v1.reset_default_graph()

    model = LXWL_name2subcategory()
    category_list = model.namelyst_predict(name_list)