Spaces:
Sleeping
Sleeping
Upload 4 files
Browse files- TCM.h5 +3 -0
- app.py +60 -0
- requirements.txt +6 -0
- scaler.joblib +3 -0
TCM.h5
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:229e71dac0ed2025ac86708b8e1ed3d48cc64b1bf01d7ea3deef1cee88ceb3ae
|
| 3 |
+
size 2665168
|
app.py
ADDED
|
@@ -0,0 +1,60 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import streamlit as st
|
| 2 |
+
import numpy as np
|
| 3 |
+
import tensorflow as tf
|
| 4 |
+
from joblib import load
|
| 5 |
+
|
| 6 |
+
scaler = load('scaler.joblib')
|
| 7 |
+
model = tf.keras.models.load_model("TCM.h5")
|
| 8 |
+
|
| 9 |
+
st.title("Cover Type Classification App")
|
| 10 |
+
|
| 11 |
+
with st.form(key='TC23'):
|
| 12 |
+
|
| 13 |
+
Elevation = st.number_input('Elevation', min_value=0, max_value=10000, value=500)
|
| 14 |
+
Aspect = st.number_input('Aspect', min_value=0, max_value=10000, value=500)
|
| 15 |
+
Slope = st.number_input('Slope', min_value=0, max_value=10000, value=500)
|
| 16 |
+
Horizontal_Distance_To_Hydrology = st.number_input('Horizontal Distance to Hydrology', min_value=0, max_value=10000, value=500)
|
| 17 |
+
Vertical_Distance_To_Hydrology = st.number_input('Vertical Distance to Hydrology', min_value=0, max_value=10000, value=500)
|
| 18 |
+
Horizontal_Distance_To_Roadways = st.number_input('Horizontal Distance to Roadways', min_value=0, max_value=10000, value=500)
|
| 19 |
+
Hillshade_9am = st.number_input('Hillshade at 9am', min_value=0, max_value=10000, value=500)
|
| 20 |
+
Hillshade_Noon = st.number_input('Hillshade at 12pm', min_value=0, max_value=10000, value=500)
|
| 21 |
+
Hillshade_3pm = st.number_input('Hillshade at 3pm', min_value=0, max_value=10000, value=500)
|
| 22 |
+
Horizontal_Distance_To_Fire_Points = st.number_input('Horizontal Distance to Fire Points', min_value=0, max_value=10000, value=500)
|
| 23 |
+
Wilderness_Area = st.number_input('Wilderness Area', min_value=1, max_value=4, value=1)
|
| 24 |
+
Soil_Type = st.number_input('Soil Type', min_value=1, max_value=40, value=1)
|
| 25 |
+
|
| 26 |
+
submitted = st.form_submit_button('Predict')
|
| 27 |
+
|
| 28 |
+
input_data = {
|
| 29 |
+
'Elevation': Elevation,
|
| 30 |
+
'Aspect': Aspect,
|
| 31 |
+
'Slope': Slope,
|
| 32 |
+
'Horizontal_Distance_To_Hydrology': Horizontal_Distance_To_Hydrology,
|
| 33 |
+
'Vertical_Distance_To_Hydrology': Vertical_Distance_To_Hydrology,
|
| 34 |
+
'Horizontal_Distance_To_Roadways': Horizontal_Distance_To_Roadways,
|
| 35 |
+
'Hillshade_9am': Hillshade_9am,
|
| 36 |
+
'Hillshade_Noon': Hillshade_Noon,
|
| 37 |
+
'Hillshade_3pm': Hillshade_3pm,
|
| 38 |
+
'Horizontal_Distance_To_Fire_Points': Horizontal_Distance_To_Fire_Points,
|
| 39 |
+
'Wilderness_Area': Wilderness_Area,
|
| 40 |
+
'Soil_Type': Soil_Type
|
| 41 |
+
}
|
| 42 |
+
|
| 43 |
+
class_to_cover_type = {
|
| 44 |
+
0: "Spruce/Fir",
|
| 45 |
+
1: "Lodgepole Pine",
|
| 46 |
+
2: "Ponderosa Pine",
|
| 47 |
+
3: "Cottonwood/Willow",
|
| 48 |
+
4: "Aspen",
|
| 49 |
+
5: "Douglas-fir",
|
| 50 |
+
6: "Krummholz",
|
| 51 |
+
}
|
| 52 |
+
|
| 53 |
+
new_test = np.array(list(input_data.values())).reshape(1, -1)
|
| 54 |
+
new_test = scaler.transform(new_test)
|
| 55 |
+
|
| 56 |
+
if submitted:
|
| 57 |
+
prediction = model.predict(new_test)
|
| 58 |
+
predicted_class = np.argmax(prediction)
|
| 59 |
+
cover_type = class_to_cover_type.get(predicted_class, "Unknown")
|
| 60 |
+
st.write(f"Predicted Cover Type: {cover_type}")
|
requirements.txt
ADDED
|
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
streamlit
|
| 2 |
+
pandas
|
| 3 |
+
numpy
|
| 4 |
+
scikit-learn
|
| 5 |
+
tensorflow
|
| 6 |
+
joblib
|
scaler.joblib
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e153166dfc1f0e6673139820e45e68c285eb8d0ec1be8240c383fabe92b35cc7
|
| 3 |
+
size 1479
|