|
|
- #!/usr/bin/env python3
-
- import util.script as script
- import util.queries as queries
- import dimod
-
- from tqdm import tqdm
-
- def main():
- wmis_siman_results_alpha_num_of_assignments()
- #wmis_siman_results()
-
- def wmis_siman_results():
- db = script.connect_to_instance_pool()
-
- q = queries.WMIS_result_scope_query_raw(db)
- q.query("c45_v[5-45]_1", "wmis_siman_results")
-
- for i in range(2):
- result = q.__next__()
-
- sample_set = queries.read_raw_wmis_sample_set(result["data"])
-
- data = script.analyze_wmis_sample(sample_set.first)
-
- print(data)
-
- def wmis_siman_results_alpha_num_of_assignments():
- edb = script.connect_to_experimetns_db()
- edb_cursor = edb.cursor()
-
- idb = script.connect_to_instance_pool()
-
- q = queries.WMIS_result_scope_query_raw(idb)
- q.query("c42_v[5-84]_1", "wmis_siman_results")
-
- insert_row = ("INSERT INTO c42_v5to84_1_wmis_results "
- "(result_id, "
- " number_of_clauses, "
- " number_of_variables, "
- " number_of_found_assignments, "
- " chain_break_fraction, "
- " num_occurrences, "
- " energy) "
- "VALUES (%s, %s, %s, %s, %s, %s, %s) ")
-
- for result in tqdm(q):
- sample_set = queries.read_raw_wmis_sample_set(result["data"])
-
- data = script.analyze_wmis_sample(sample_set.first)
-
- sat = queries.get_instance_by_id(idb["instances"], result["instance"])
-
- edb_cursor.execute(insert_row, (str(result["_id"]),
- int(sat.getNumberOfClauses()),
- int(sat.getNumberOfVariables()),
- int(data["number_of_assignments"]),
- float(data["chain_break_fraction"]),
- int(data["num_occurrences"]),
- int(data["energy"])))
-
- edb.commit()
- edb_cursor.close()
- edb.close()
-
- if __name__ == "__main__":
- main()
|
