feature: add decision tree
This commit is contained in:
parent
37c5489403
commit
a5d8ab61a6
@ -9,7 +9,7 @@ from sklearn.tree import export_graphviz
|
|||||||
import pydotplus
|
import pydotplus
|
||||||
|
|
||||||
# Read the dataset
|
# Read the dataset
|
||||||
train_data_m = pd.read_csv("datasets/treedataset2.csv")
|
train_data_m = pd.read_csv("datasets/bees.csv")
|
||||||
|
|
||||||
feature_cols = ['Weather','Temperature','WindSpeed','PollenAvailability','BeePopulation','TimeOfDay','FlowerDistance','Humidity']
|
feature_cols = ['Weather','Temperature','WindSpeed','PollenAvailability','BeePopulation','TimeOfDay','FlowerDistance','Humidity']
|
||||||
print(train_data_m.head())
|
print(train_data_m.head())
|
||||||
@ -37,4 +37,3 @@ export_graphviz(clf, out_file=dot_data,
|
|||||||
special_characters=True, feature_names = feature_cols,class_names=['0','1'])
|
special_characters=True, feature_names = feature_cols,class_names=['0','1'])
|
||||||
graph = pydotplus.graph_from_dot_data(dot_data.getvalue())
|
graph = pydotplus.graph_from_dot_data(dot_data.getvalue())
|
||||||
graph.write_png('tree.png')
|
graph.write_png('tree.png')
|
||||||
Image(graph.create_png())
|
|
@ -1,4 +1,4 @@
|
|||||||
Weather,Temperature,Wind Speed,Pollen Availability,Bee Population,Time of Day,Flower Distance,Humidity,Fly_out_for_honey
|
Weather,Temperature,WindSpeed,PollenAvailability,BeePopulation,TimeOfDay,FlowerDistance,Humidity,FlyOutForHoney
|
||||||
Cloudy,Medium,Strong,Low,High,Morning,Close,Normal,No
|
Cloudy,Medium,Strong,Low,High,Morning,Close,Normal,No
|
||||||
Cloudy,Low,Strong,High,Moderate,Morning,Close,High,Yes
|
Cloudy,Low,Strong,High,Moderate,Morning,Close,High,Yes
|
||||||
Sunny,High,Strong,Medium,Moderate,Morning,Close,Normal,Yes
|
Sunny,High,Strong,Medium,Moderate,Morning,Close,Normal,Yes
|
||||||
|
|
@ -1,4 +1,4 @@
|
|||||||
Weather,Temperature,Wind Speed,Pollen Availability,Bee Population,Time of Day,Flower Distance,Humidity,Fly_out_for_honey
|
Weather,Temperature,WindSpeed,PollenAvailability,BeePopulation,TimeOfDay,FlowerDistance,Humidity,FlyOutForHoney
|
||||||
Cloudy,High,Strong,Medium,Moderate,Evening,Close,High,Yes
|
Cloudy,High,Strong,Medium,Moderate,Evening,Close,High,Yes
|
||||||
Rainy,Low,Strong,Medium,Moderate,Afternoon,Medium,High,No
|
Rainy,Low,Strong,Medium,Moderate,Afternoon,Medium,High,No
|
||||||
Cloudy,Medium,Strong,Medium,Moderate,Evening,Medium,Normal,Yes
|
Cloudy,Medium,Strong,Medium,Moderate,Evening,Medium,Normal,Yes
|
||||||
|
|
@ -100,7 +100,6 @@ bee_dir = 'west'
|
|||||||
field = Field(tilemap_types, tilemap_values)
|
field = Field(tilemap_types, tilemap_values)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def Manhattan_dis(start, end):
|
def Manhattan_dis(start, end):
|
||||||
start_x, start_y = start
|
start_x, start_y = start
|
||||||
end_x, end_y = end
|
end_x, end_y = end
|
||||||
|
BIN
src/tree.png
BIN
src/tree.png
Binary file not shown.
Before Width: | Height: | Size: 662 KiB After Width: | Height: | Size: 690 KiB |
Loading…
Reference in New Issue
Block a user