decision tree source
This commit is contained in:
parent
2040f37cd8
commit
2115b21f46
BIN
DecisionTree/IMG_20230525_193651.jpg
Normal file
BIN
DecisionTree/IMG_20230525_193651.jpg
Normal file
Binary file not shown.
After Width: | Height: | Size: 504 KiB |
84
DecisionTree/decision_based_on_table_script.py
Normal file
84
DecisionTree/decision_based_on_table_script.py
Normal file
@ -0,0 +1,84 @@
|
||||
def read_text_document(file_path): # read given amount of lines (lines of 8 integers)
|
||||
string_array = []
|
||||
|
||||
try:
|
||||
with open(file_path, 'r') as file:
|
||||
# Read each line of the document
|
||||
for line in file:
|
||||
# Remove trailing newline characters and append to the array
|
||||
string_array.append(line.rstrip('\n'))
|
||||
except FileNotFoundError:
|
||||
print("File not found.")
|
||||
|
||||
return string_array # returns array of lines from the file
|
||||
|
||||
|
||||
def gen_output(array, second_array): # transcribes line of integers into elemental decisions of shelf "g" and "d"
|
||||
if array[0] == 0 or array[0] == 1:
|
||||
second_array[0] = "g"
|
||||
else:
|
||||
second_array[0] = "d"
|
||||
if array[1] == 1 or array[1] == 2:
|
||||
second_array[1] = "d"
|
||||
else:
|
||||
second_array[1] = "g"
|
||||
if array[2] == 0:
|
||||
second_array[2] = "g"
|
||||
else:
|
||||
second_array[2] = "d"
|
||||
if array[3] == 0:
|
||||
second_array[3] = "g"
|
||||
else:
|
||||
second_array[3] = "d"
|
||||
if array[4] == 0:
|
||||
second_array[4] = "d"
|
||||
else:
|
||||
second_array[4] = "g"
|
||||
if array[5] == 0:
|
||||
second_array[5] = "g"
|
||||
else:
|
||||
second_array[5] = "d"
|
||||
if array[6] == 0:
|
||||
second_array[6] = "d"
|
||||
else:
|
||||
second_array[6] = "g"
|
||||
if array[7] == 0:
|
||||
second_array[7] = "d"
|
||||
else:
|
||||
second_array[7] = "g"
|
||||
|
||||
def count(array): # count number of g and d and make decision, if same number return 2 instead
|
||||
d = 0
|
||||
g = 0
|
||||
for digit in array:
|
||||
if digit == "g":
|
||||
g += 1
|
||||
else:
|
||||
d += 1
|
||||
if d > g:
|
||||
return 0 # lower shelf
|
||||
elif g > d:
|
||||
return 1 # upper shelf
|
||||
else:
|
||||
return 2 # optimisation of space, goes to more empty shelf overall
|
||||
|
||||
file_path = 'file/path/to/input/lines/of/integers'
|
||||
examples = read_text_document(file_path) # array of given number of examples from file
|
||||
|
||||
for input in examples:
|
||||
digit_array = []
|
||||
|
||||
for char in input:
|
||||
# Convert the character to an integer and add it to the array
|
||||
digit_array.append(int(char))
|
||||
|
||||
output_array = [None] * 8
|
||||
gen_output(digit_array, output_array)
|
||||
|
||||
decision_output = count(output_array)
|
||||
if decision_output == 2: # in case d == g, check which shelf is more empty
|
||||
if output_array[7] == "g":
|
||||
decision_output = 1
|
||||
elif output_array[7] == "d":
|
||||
decision_output = 0
|
||||
print(decision_output) # final decision
|
Loading…
Reference in New Issue
Block a user