add parameters for decision tree

This commit is contained in:
matixezor 2021-05-17 17:53:08 +02:00
parent 5560d6687d
commit 13ccc6c184
5 changed files with 30 additions and 9 deletions

10
src/adm_mine.py Normal file
View File

@ -0,0 +1,10 @@
from typing import Union
from mine import Mine
class ADMMine(Mine):
mine_type = 'ADM'
def __init__(self, pressure: Union[float, int], armed: bool = True):
super().__init__(pressure, armed)

View File

@ -1,8 +1,10 @@
from typing import Union
from mine import Mine from mine import Mine
class APMine(Mine): class APMine(Mine):
defuse_method = 2 mine_type = 'AP'
def __init__(self): def __init__(self, pressure: Union[float, int], armed: bool = True):
super().__init__() super().__init__(pressure, armed)

View File

@ -1,8 +1,10 @@
from typing import Union
from mine import Mine from mine import Mine
class ATMine(Mine): class ATMine(Mine):
defuse_method = 1 mine_type = 'AT'
def __init__(self): def __init__(self, pressure: Union[float, int], armed: bool = True):
super().__init__() super().__init__(pressure, armed)

View File

@ -1,3 +1,7 @@
from typing import Union
class Mine: class Mine:
def __init__(self, is_armed: bool = True): def __init__(self, pressure: Union[float, int], armed: bool):
self.is_armed = is_armed self.is_armed = armed
self.pressure = pressure

View File

@ -5,7 +5,10 @@ from at_mine import ATMine
class Tile: class Tile:
def __init__(self, number: int, weight: int, mine: Union[None, APMine, ATMine] = None): def __init__(self, number: int, weight: int, visibility, stability, ground, mine: Union[None, APMine, ATMine] = None):
self.number = number self.number = number
self.mine = mine self.mine = mine
self.weight = weight self.weight = weight
self.visibility = visibility
self.stability = stability
self.ground = ground