Skip to content
Projects
Groups
Snippets
Help
Loading...
Sign in
Toggle navigation
F
ffm-baseline
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
ML
ffm-baseline
Commits
c113a6a8
Commit
c113a6a8
authored
Aug 22, 2018
by
张彦钊
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
add print
parent
0b7df7a5
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
11 additions
and
11 deletions
+11
-11
diaryQueueUpdate.py
diaryQueueUpdate.py
+11
-11
No files found.
diaryQueueUpdate.py
View file @
c113a6a8
...
...
@@ -43,7 +43,7 @@ def test_con_sql(device_id):
megacity_queue
=
list
(
set
(
megacity_queue
)
&
set
(
data_set_cid
))
db
.
close
()
print
(
"成功获取日记队列"
)
#
print("成功获取日记队列")
return
native_queue
,
nearby_queue
,
nation_queue
,
megacity_queue
else
:
...
...
@@ -64,7 +64,7 @@ def get_native_queue(device_id):
native_queue
=
list
(
map
(
lambda
x
:
"diary|"
+
str
(
x
),
native_queue
))
native_queue
=
list
(
set
(
native_queue
)
&
set
(
data_set_cid
))
db
.
close
()
print
(
"成功获取native_queue"
)
#
print("成功获取native_queue")
return
native_queue
else
:
return
None
...
...
@@ -85,7 +85,7 @@ def feature_en(x_list, device_id):
data
[
"minute"
]
=
data
[
"minute"
]
.
astype
(
"category"
)
# 虽然预测y,但ffm转化需要y,并不影响预测结果
data
[
"y"
]
=
0
print
(
"done 特征工程"
)
#
print("done 特征工程")
return
data
...
...
@@ -97,7 +97,7 @@ def transform_ffm_format(df,queue_name):
data
=
ffm_format_pandas
.
transform
(
df
)
predict_file_name
=
DIRECTORY_PATH
+
"result/{0}_{1}.csv"
.
format
(
device_id
,
queue_name
)
data
.
to_csv
(
predict_file_name
,
index
=
False
,
header
=
None
)
print
(
"done ffm"
)
#
print("done ffm")
return
predict_file_name
...
...
@@ -112,19 +112,19 @@ def predict(queue_name, x_list):
ffm_model
.
predict
(
DIRECTORY_PATH
+
"model.out"
,
DIRECTORY_PATH
+
"result/output{0}_{1}.csv"
.
format
(
device_id
,
queue_name
))
print
(
"done predict"
)
#
print("done predict")
def
save_result
(
queue_name
,
x_list
):
score_df
=
pd
.
read_csv
(
DIRECTORY_PATH
+
"result/output{0}_{1}.csv"
.
format
(
device_id
,
queue_name
),
header
=
None
)
print
(
score_df
)
#
print(score_df)
mm_scaler
=
MinMaxScaler
()
mm_scaler
.
fit
(
score_df
)
score_df
=
pd
.
DataFrame
(
mm_scaler
.
transform
(
score_df
))
print
(
"概率前十行:"
)
print
(
score_df
)
#
print("概率前十行:")
#
print(score_df)
score_df
=
score_df
.
rename
(
columns
=
{
0
:
"score"
})
score_df
[
"cid"
]
=
x_list
...
...
@@ -198,7 +198,7 @@ def update_sql_dairy_queue(queue_name, diary_id):
sql
=
"update device_diary_queue set {}='{}' where device_id = '{}'"
.
format
(
queue_name
,
diary_id
,
device_id
)
cursor
.
execute
(
sql
)
db
.
close
()
print
(
"成功写入"
)
#
print("成功写入")
def
multi_update
(
key
,
name_dict
):
...
...
@@ -209,9 +209,9 @@ def multi_update(key, name_dict):
if
get_native_queue
(
device_id
)
==
native_queue_list
:
update_sql_dairy_queue
(
key
,
diary_id
)
print
(
"更新结束"
)
#
print("更新结束")
else
:
print
(
"不需要更新日记队列"
)
#
print("不需要更新日记队列")
if
__name__
==
"__main__"
:
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment