|
|
@@ -225,7 +225,7 @@ def generate_push_dataset():
|
|
|
data_set = mysql_client.select(fetch_query, cursor_type=DictCursor)
|
|
|
filter_conversation = [i for i in data_set if len(json.loads(i['conversation'])) >= 20]
|
|
|
|
|
|
- samples =random.sample(filter_conversation, 100)
|
|
|
+ samples =random.sample(filter_conversation, 300)
|
|
|
|
|
|
# init message push agent
|
|
|
for sample in tqdm(samples):
|
luojunhui