extract_experiment.py 931 B

1234567891011121314151617181920212223242526272829303132333435
  1. # db connection
  2. from pymongo import MongoClient
  3. import json, os
  4. connection = MongoClient('mongodb://mongo:27017')
  5. db = connection['sin3d']
  6. data_collection = db['datas']
  7. output_results_folder = 'results'
  8. experiments_identifier = ['sin3d-PrISE-3D']
  9. experiment_results = data_collection.find({
  10. 'data.msg.experimentName': 'MatchExtractsWithReference',
  11. 'data.msgId': 'EXPERIMENT_VALIDATED',
  12. 'data.experimentId':{
  13. '$in': experiments_identifier
  14. }
  15. # '$not': { '$gt': 1.99 }
  16. })
  17. if not os.path.exists(output_results_folder):
  18. os.makedirs(output_results_folder)
  19. results_filename = 'experiments_results.json'
  20. results_filepath = os.path.join(output_results_folder, results_filename)
  21. export_data = []
  22. for result in experiment_results:
  23. export_data.append(result['data'])
  24. print('Save results into', results_filepath)
  25. with open(results_filepath, 'w') as f:
  26. f.write(json.dumps(export_data, indent=4))