Browse Source

初始化

罗俊辉 1 year ago
parent
commit
d9ac747ed7
1 changed files with 5 additions and 4 deletions
  1. 5 4
      applications/functions.py

+ 5 - 4
applications/functions.py

@@ -30,14 +30,15 @@ class ParamProcess(object):
             if title_tags:
                 for i in range(3):
                     try:
-                        features['tag_{}'.format(i + 1)] = title_tags[i]
+                        features['tag{}'.format(i + 1)] = title_tags[i]
                     except:
                         features['tag_{}'.format(i + 1)] = None
             else:
-                features['tag_1'] = None
-                features['tag_2'] = None
-                features['tag_3'] = None
+                features['tag1'] = None
+                features['tag2'] = None
+                features['tag3'] = None
         df = pd.DataFrame([features])
+        print("data_frame", df.columns)
         df.drop('title', axis=1)
         return df