|
@@ -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
|
|
|
|