changed model format to .h5

This commit is contained in:
Maks Kulikowski 2022-06-08 18:17:47 +02:00
parent f9ea288da1
commit 7439ccdff9
6 changed files with 31 additions and 31 deletions

Binary file not shown.

Before

Width:  |  Height:  |  Size: 20 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 134 KiB

After

Width:  |  Height:  |  Size: 20 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 134 KiB

View File

@ -13,8 +13,8 @@ warnings.filterwarnings('ignore')
create_model = False
learning_sets_path = "data/learning_sets"
save_model_path = "data/models/true_mine_recognizer.model"
load_model_path = "data/models/true_mine_recognizer.model"
save_model_path = "data/models/true_mine_recognizer2.h5"
load_model_path = "data/models/true_mine_recognizer2.h5"
image_size = 128
class NeuralNetwork():

Binary file not shown.

View File

@ -1,31 +1,31 @@
[
{"current_level": 1, "leaf_id": "a5207776-e038-11ec-81fa-d43d7ef1576e", "parents": "root", "rule": "else: return 'detonate'", "feature_idx": -1, "feature_name": "", "instances": 200, "metric": 0, "return_statement": 0, "tree_id": 0},
{"current_level": 1, "leaf_id": "a52161b8-e038-11ec-a09a-d43d7ef1576e", "parents": "root", "rule": "if obj[4]<=80.67436609605278:", "feature_idx": 4, "feature_name": "Random_detonation_chance", "instances": 200, "metric": 0.9964625048848765, "return_statement": 0, "tree_id": 0},
{"current_level": 2, "leaf_id": "a5307c65-e038-11ec-b18b-d43d7ef1576e", "parents": "a52161b8-e038-11ec-a09a-d43d7ef1576e", "rule": "else: return 'detonate'", "feature_idx": -1, "feature_name": "", "instances": 161, "metric": 0, "return_statement": 0, "tree_id": 0},
{"current_level": 2, "leaf_id": "a5318dc0-e038-11ec-986d-d43d7ef1576e", "parents": "a52161b8-e038-11ec-a09a-d43d7ef1576e", "rule": "if obj[2]<=0:", "feature_idx": 2, "feature_name": "Protection_from_defuse", "instances": 161, "metric": 0.9203523274205176, "return_statement": 0, "tree_id": 0},
{"current_level": 3, "leaf_id": "a53f21ad-e038-11ec-a443-d43d7ef1576e", "parents": "a5318dc0-e038-11ec-986d-d43d7ef1576e", "rule": "else: return 'detonate'", "feature_idx": -1, "feature_name": "", "instances": 135, "metric": 0, "return_statement": 0, "tree_id": 0},
{"current_level": 3, "leaf_id": "a5403290-e038-11ec-9ad7-d43d7ef1576e", "parents": "a5318dc0-e038-11ec-986d-d43d7ef1576e", "rule": "if obj[3]<=7:", "feature_idx": 3, "feature_name": "Meters_under_the_ground", "instances": 135, "metric": 0.7364977795505669, "return_statement": 0, "tree_id": 0},
{"current_level": 4, "leaf_id": "a549f58f-e038-11ec-9721-d43d7ef1576e", "parents": "a5403290-e038-11ec-9ad7-d43d7ef1576e", "rule": "else: return 'defuse'", "feature_idx": -1, "feature_name": "", "instances": 116, "metric": 0, "return_statement": 0, "tree_id": 0},
{"current_level": 4, "leaf_id": "a54adfdd-e038-11ec-ab1c-d43d7ef1576e", "parents": "a5403290-e038-11ec-9ad7-d43d7ef1576e", "rule": "if obj[1]<=1997.8794790831414:", "feature_idx": 1, "feature_name": "Year(older_more_difficult)", "instances": 116, "metric": 0.3936164041111624, "return_statement": 0, "tree_id": 0},
{"current_level": 5, "leaf_id": "a5525982-e038-11ec-a259-d43d7ef1576e", "parents": "a54adfdd-e038-11ec-ab1c-d43d7ef1576e", "rule": "else: return 'defuse'", "feature_idx": -1, "feature_name": "", "instances": 97, "metric": 0, "return_statement": 0, "tree_id": 0},
{"current_level": 5, "leaf_id": "a55343a5-e038-11ec-8973-d43d7ef1576e", "parents": "a54adfdd-e038-11ec-ab1c-d43d7ef1576e", "rule": "if obj[5]<=2:", "feature_idx": 5, "feature_name": "Detonation_power_in_m", "instances": 97, "metric": 0.445693177722561, "return_statement": 0, "tree_id": 0},
{"current_level": 6, "leaf_id": "a55713e5-e038-11ec-b018-d43d7ef1576e", "parents": "a55343a5-e038-11ec-8973-d43d7ef1576e", "rule": "else: return 'defuse'", "feature_idx": -1, "feature_name": "", "instances": 65, "metric": 0, "return_statement": 0, "tree_id": 0},
{"current_level": 6, "leaf_id": "a55824b5-e038-11ec-a34f-d43d7ef1576e", "parents": "a55343a5-e038-11ec-8973-d43d7ef1576e", "rule": "if obj[0]>3:", "feature_idx": 0, "feature_name": "Size(bigger_more_difficult)", "instances": 65, "metric": 0.33352200393097664, "return_statement": 0, "tree_id": 0},
{"current_level": 7, "leaf_id": "a55824b6-e038-11ec-bb03-d43d7ef1576e", "parents": "a55824b5-e038-11ec-a34f-d43d7ef1576e", "rule": "return 'defuse'", "feature_idx": 0, "feature_name": "Size(bigger_more_difficult)", "instances": 65, "metric": 0, "return_statement": 1, "tree_id": 0},
{"current_level": 6, "leaf_id": "a564f4e5-e038-11ec-a139-d43d7ef1576e", "parents": "a55343a5-e038-11ec-8973-d43d7ef1576e", "rule": "if obj[0]<=3:", "feature_idx": 0, "feature_name": "Size(bigger_more_difficult)", "instances": 65, "metric": 0.33352200393097664, "return_statement": 0, "tree_id": 0},
{"current_level": 7, "leaf_id": "a564f4e6-e038-11ec-b59b-d43d7ef1576e", "parents": "a564f4e5-e038-11ec-a139-d43d7ef1576e", "rule": "return 'defuse'", "feature_idx": 0, "feature_name": "Size(bigger_more_difficult)", "instances": 65, "metric": 0, "return_statement": 1, "tree_id": 0},
{"current_level": 5, "leaf_id": "a57d103d-e038-11ec-837c-d43d7ef1576e", "parents": "a54adfdd-e038-11ec-ab1c-d43d7ef1576e", "rule": "if obj[5]>2:", "feature_idx": 5, "feature_name": "Detonation_power_in_m", "instances": 97, "metric": 0.445693177722561, "return_statement": 0, "tree_id": 0},
{"current_level": 6, "leaf_id": "a580b941-e038-11ec-8b02-d43d7ef1576e", "parents": "a57d103d-e038-11ec-837c-d43d7ef1576e", "rule": "else: return 'defuse'", "feature_idx": -1, "feature_name": "", "instances": 32, "metric": 0, "return_statement": 0, "tree_id": 0},
{"current_level": 6, "leaf_id": "a581cac7-e038-11ec-9903-d43d7ef1576e", "parents": "a57d103d-e038-11ec-837c-d43d7ef1576e", "rule": "if obj[0]<=7:", "feature_idx": 0, "feature_name": "Size(bigger_more_difficult)", "instances": 32, "metric": 0.6252624052234231, "return_statement": 0, "tree_id": 0},
{"current_level": 7, "leaf_id": "a581cac8-e038-11ec-9a27-d43d7ef1576e", "parents": "a581cac7-e038-11ec-9903-d43d7ef1576e", "rule": "return 'defuse'", "feature_idx": 0, "feature_name": "Size(bigger_more_difficult)", "instances": 32, "metric": 0, "return_statement": 1, "tree_id": 0},
{"current_level": 6, "leaf_id": "a58dd70d-e038-11ec-9d70-d43d7ef1576e", "parents": "a57d103d-e038-11ec-837c-d43d7ef1576e", "rule": "if obj[0]>7:", "feature_idx": 0, "feature_name": "Size(bigger_more_difficult)", "instances": 32, "metric": 0.6252624052234231, "return_statement": 0, "tree_id": 0},
{"current_level": 7, "leaf_id": "a58dd70e-e038-11ec-a5cf-d43d7ef1576e", "parents": "a58dd70d-e038-11ec-9d70-d43d7ef1576e", "rule": "return 'defuse'", "feature_idx": 0, "feature_name": "Size(bigger_more_difficult)", "instances": 32, "metric": 0, "return_statement": 1, "tree_id": 0},
{"current_level": 4, "leaf_id": "a5b27499-e038-11ec-9c75-d43d7ef1576e", "parents": "a5403290-e038-11ec-9ad7-d43d7ef1576e", "rule": "if obj[1]>1997.8794790831414:", "feature_idx": 1, "feature_name": "Year(older_more_difficult)", "instances": 116, "metric": 0.3936164041111624, "return_statement": 0, "tree_id": 0},
{"current_level": 5, "leaf_id": "a5b2749a-e038-11ec-ad1c-d43d7ef1576e", "parents": "a5b27499-e038-11ec-9c75-d43d7ef1576e", "rule": "return 'defuse'", "feature_idx": 1, "feature_name": "Year(older_more_difficult)", "instances": 116, "metric": 0, "return_statement": 1, "tree_id": 0},
{"current_level": 3, "leaf_id": "a5cadc0d-e038-11ec-ab67-d43d7ef1576e", "parents": "a5318dc0-e038-11ec-986d-d43d7ef1576e", "rule": "if obj[3]>7:", "feature_idx": 3, "feature_name": "Meters_under_the_ground", "instances": 135, "metric": 0.7364977795505669, "return_statement": 0, "tree_id": 0},
{"current_level": 4, "leaf_id": "a5cadc0e-e038-11ec-bd41-d43d7ef1576e", "parents": "a5cadc0d-e038-11ec-ab67-d43d7ef1576e", "rule": "return 'detonate'", "feature_idx": 3, "feature_name": "Meters_under_the_ground", "instances": 135, "metric": 0, "return_statement": 1, "tree_id": 0},
{"current_level": 2, "leaf_id": "a5e31ebf-e038-11ec-9eb3-d43d7ef1576e", "parents": "a52161b8-e038-11ec-a09a-d43d7ef1576e", "rule": "if obj[2]>0:", "feature_idx": 2, "feature_name": "Protection_from_defuse", "instances": 161, "metric": 0.9203523274205176, "return_statement": 0, "tree_id": 0},
{"current_level": 3, "leaf_id": "a5e31ec0-e038-11ec-86a5-d43d7ef1576e", "parents": "a5e31ebf-e038-11ec-9eb3-d43d7ef1576e", "rule": "return 'detonate'", "feature_idx": 2, "feature_name": "Protection_from_defuse", "instances": 161, "metric": 0, "return_statement": 1, "tree_id": 0},
{"current_level": 1, "leaf_id": "a5fb1242-e038-11ec-bb21-d43d7ef1576e", "parents": "root", "rule": "if obj[4]>80.67436609605278:", "feature_idx": 4, "feature_name": "Random_detonation_chance", "instances": 200, "metric": 0.9964625048848765, "return_statement": 0, "tree_id": 0},
{"current_level": 2, "leaf_id": "a5fb1243-e038-11ec-9433-d43d7ef1576e", "parents": "a5fb1242-e038-11ec-bb21-d43d7ef1576e", "rule": "return 'detonate'", "feature_idx": 4, "feature_name": "Random_detonation_chance", "instances": 200, "metric": 0, "return_statement": 1, "tree_id": 0}
{"current_level": 1, "leaf_id": "35029f9c-e746-11ec-b6a5-d43d7ef1576e", "parents": "root", "rule": "else: return 'detonate'", "feature_idx": -1, "feature_name": "", "instances": 200, "metric": 0, "return_statement": 0, "tree_id": 0},
{"current_level": 1, "leaf_id": "3503d820-e746-11ec-86d2-d43d7ef1576e", "parents": "root", "rule": "if obj[4]<=80.67436609605278:", "feature_idx": 4, "feature_name": "Random_detonation_chance", "instances": 200, "metric": 0.9964625048848765, "return_statement": 0, "tree_id": 0},
{"current_level": 2, "leaf_id": "3511e0ba-e746-11ec-a377-d43d7ef1576e", "parents": "3503d820-e746-11ec-86d2-d43d7ef1576e", "rule": "else: return 'detonate'", "feature_idx": -1, "feature_name": "", "instances": 161, "metric": 0, "return_statement": 0, "tree_id": 0},
{"current_level": 2, "leaf_id": "3512f36a-e746-11ec-b5ed-d43d7ef1576e", "parents": "3503d820-e746-11ec-86d2-d43d7ef1576e", "rule": "if obj[2]<=0:", "feature_idx": 2, "feature_name": "Protection_from_defuse", "instances": 161, "metric": 0.9203523274205176, "return_statement": 0, "tree_id": 0},
{"current_level": 3, "leaf_id": "3520380e-e746-11ec-9429-d43d7ef1576e", "parents": "3512f36a-e746-11ec-b5ed-d43d7ef1576e", "rule": "else: return 'detonate'", "feature_idx": -1, "feature_name": "", "instances": 135, "metric": 0, "return_statement": 0, "tree_id": 0},
{"current_level": 3, "leaf_id": "35217079-e746-11ec-bc97-d43d7ef1576e", "parents": "3512f36a-e746-11ec-b5ed-d43d7ef1576e", "rule": "if obj[3]<=7:", "feature_idx": 3, "feature_name": "Meters_under_the_ground", "instances": 135, "metric": 0.7364977795505669, "return_statement": 0, "tree_id": 0},
{"current_level": 4, "leaf_id": "352b3a9d-e746-11ec-ab0d-d43d7ef1576e", "parents": "35217079-e746-11ec-bc97-d43d7ef1576e", "rule": "else: return 'defuse'", "feature_idx": -1, "feature_name": "", "instances": 116, "metric": 0, "return_statement": 0, "tree_id": 0},
{"current_level": 4, "leaf_id": "352c73ad-e746-11ec-a3f8-d43d7ef1576e", "parents": "35217079-e746-11ec-bc97-d43d7ef1576e", "rule": "if obj[1]<=1997.8794790831414:", "feature_idx": 1, "feature_name": "Year(older_more_difficult)", "instances": 116, "metric": 0.3936164041111624, "return_statement": 0, "tree_id": 0},
{"current_level": 5, "leaf_id": "353350bc-e746-11ec-85c4-d43d7ef1576e", "parents": "352c73ad-e746-11ec-a3f8-d43d7ef1576e", "rule": "else: return 'defuse'", "feature_idx": -1, "feature_name": "", "instances": 97, "metric": 0, "return_statement": 0, "tree_id": 0},
{"current_level": 5, "leaf_id": "3534620e-e746-11ec-8d75-d43d7ef1576e", "parents": "352c73ad-e746-11ec-a3f8-d43d7ef1576e", "rule": "if obj[5]<=2:", "feature_idx": 5, "feature_name": "Detonation_power_in_m", "instances": 97, "metric": 0.445693177722561, "return_statement": 0, "tree_id": 0},
{"current_level": 6, "leaf_id": "35383282-e746-11ec-8d3e-d43d7ef1576e", "parents": "3534620e-e746-11ec-8d75-d43d7ef1576e", "rule": "else: return 'defuse'", "feature_idx": -1, "feature_name": "", "instances": 65, "metric": 0, "return_statement": 0, "tree_id": 0},
{"current_level": 6, "leaf_id": "35396ab6-e746-11ec-ae2f-d43d7ef1576e", "parents": "3534620e-e746-11ec-8d75-d43d7ef1576e", "rule": "if obj[0]>3:", "feature_idx": 0, "feature_name": "Size(bigger_more_difficult)", "instances": 65, "metric": 0.33352200393097664, "return_statement": 0, "tree_id": 0},
{"current_level": 7, "leaf_id": "35396ab7-e746-11ec-96e9-d43d7ef1576e", "parents": "35396ab6-e746-11ec-ae2f-d43d7ef1576e", "rule": "return 'defuse'", "feature_idx": 0, "feature_name": "Size(bigger_more_difficult)", "instances": 65, "metric": 0, "return_statement": 1, "tree_id": 0},
{"current_level": 6, "leaf_id": "35450200-e746-11ec-a100-d43d7ef1576e", "parents": "3534620e-e746-11ec-8d75-d43d7ef1576e", "rule": "if obj[0]<=3:", "feature_idx": 0, "feature_name": "Size(bigger_more_difficult)", "instances": 65, "metric": 0.33352200393097664, "return_statement": 0, "tree_id": 0},
{"current_level": 7, "leaf_id": "35450201-e746-11ec-b6d8-d43d7ef1576e", "parents": "35450200-e746-11ec-a100-d43d7ef1576e", "rule": "return 'defuse'", "feature_idx": 0, "feature_name": "Size(bigger_more_difficult)", "instances": 65, "metric": 0, "return_statement": 1, "tree_id": 0},
{"current_level": 5, "leaf_id": "355be37b-e746-11ec-8934-d43d7ef1576e", "parents": "352c73ad-e746-11ec-a3f8-d43d7ef1576e", "rule": "if obj[5]>2:", "feature_idx": 5, "feature_name": "Detonation_power_in_m", "instances": 97, "metric": 0.445693177722561, "return_statement": 0, "tree_id": 0},
{"current_level": 6, "leaf_id": "355f659c-e746-11ec-b1cb-d43d7ef1576e", "parents": "355be37b-e746-11ec-8934-d43d7ef1576e", "rule": "else: return 'defuse'", "feature_idx": -1, "feature_name": "", "instances": 32, "metric": 0, "return_statement": 0, "tree_id": 0},
{"current_level": 6, "leaf_id": "35609e04-e746-11ec-a6f5-d43d7ef1576e", "parents": "355be37b-e746-11ec-8934-d43d7ef1576e", "rule": "if obj[0]<=7:", "feature_idx": 0, "feature_name": "Size(bigger_more_difficult)", "instances": 32, "metric": 0.6252624052234231, "return_statement": 0, "tree_id": 0},
{"current_level": 7, "leaf_id": "35609e05-e746-11ec-a26b-d43d7ef1576e", "parents": "35609e04-e746-11ec-a6f5-d43d7ef1576e", "rule": "return 'defuse'", "feature_idx": 0, "feature_name": "Size(bigger_more_difficult)", "instances": 32, "metric": 0, "return_statement": 1, "tree_id": 0},
{"current_level": 6, "leaf_id": "356c0f3c-e746-11ec-ae44-d43d7ef1576e", "parents": "355be37b-e746-11ec-8934-d43d7ef1576e", "rule": "if obj[0]>7:", "feature_idx": 0, "feature_name": "Size(bigger_more_difficult)", "instances": 32, "metric": 0.6252624052234231, "return_statement": 0, "tree_id": 0},
{"current_level": 7, "leaf_id": "356c0f3d-e746-11ec-95c6-d43d7ef1576e", "parents": "356c0f3c-e746-11ec-ae44-d43d7ef1576e", "rule": "return 'defuse'", "feature_idx": 0, "feature_name": "Size(bigger_more_difficult)", "instances": 32, "metric": 0, "return_statement": 1, "tree_id": 0},
{"current_level": 4, "leaf_id": "358ea0ff-e746-11ec-8643-d43d7ef1576e", "parents": "35217079-e746-11ec-bc97-d43d7ef1576e", "rule": "if obj[1]>1997.8794790831414:", "feature_idx": 1, "feature_name": "Year(older_more_difficult)", "instances": 116, "metric": 0.3936164041111624, "return_statement": 0, "tree_id": 0},
{"current_level": 5, "leaf_id": "358ea100-e746-11ec-8ec1-d43d7ef1576e", "parents": "358ea0ff-e746-11ec-8643-d43d7ef1576e", "rule": "return 'defuse'", "feature_idx": 1, "feature_name": "Year(older_more_difficult)", "instances": 116, "metric": 0, "return_statement": 1, "tree_id": 0},
{"current_level": 3, "leaf_id": "35a58276-e746-11ec-9de5-d43d7ef1576e", "parents": "3512f36a-e746-11ec-b5ed-d43d7ef1576e", "rule": "if obj[3]>7:", "feature_idx": 3, "feature_name": "Meters_under_the_ground", "instances": 135, "metric": 0.7364977795505669, "return_statement": 0, "tree_id": 0},
{"current_level": 4, "leaf_id": "35a58277-e746-11ec-a89b-d43d7ef1576e", "parents": "35a58276-e746-11ec-9de5-d43d7ef1576e", "rule": "return 'detonate'", "feature_idx": 3, "feature_name": "Meters_under_the_ground", "instances": 135, "metric": 0, "return_statement": 1, "tree_id": 0},
{"current_level": 2, "leaf_id": "35bcb2be-e746-11ec-867f-d43d7ef1576e", "parents": "3503d820-e746-11ec-86d2-d43d7ef1576e", "rule": "if obj[2]>0:", "feature_idx": 2, "feature_name": "Protection_from_defuse", "instances": 161, "metric": 0.9203523274205176, "return_statement": 0, "tree_id": 0},
{"current_level": 3, "leaf_id": "35bcb2bf-e746-11ec-8e16-d43d7ef1576e", "parents": "35bcb2be-e746-11ec-867f-d43d7ef1576e", "rule": "return 'detonate'", "feature_idx": 2, "feature_name": "Protection_from_defuse", "instances": 161, "metric": 0, "return_statement": 1, "tree_id": 0},
{"current_level": 1, "leaf_id": "35d36d2a-e746-11ec-ae96-d43d7ef1576e", "parents": "root", "rule": "if obj[4]>80.67436609605278:", "feature_idx": 4, "feature_name": "Random_detonation_chance", "instances": 200, "metric": 0.9964625048848765, "return_statement": 0, "tree_id": 0},
{"current_level": 2, "leaf_id": "35d39406-e746-11ec-8e91-d43d7ef1576e", "parents": "35d36d2a-e746-11ec-ae96-d43d7ef1576e", "rule": "return 'detonate'", "feature_idx": 4, "feature_name": "Random_detonation_chance", "instances": 200, "metric": 0, "return_statement": 1, "tree_id": 0}
]