Fixing tensorflow warning

lwdovico · Aug 12, 2022 · 3d566ff · 3d566ff
1 parent 852a470
commit 3d566ff
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/price_predictor/price_predictor.py b/price_predictor/price_predictor.py
@@ -11,6 +11,7 @@
 
 from sklearn.preprocessing import MinMaxScaler
 
+os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
 from tensorflow.keras.models import load_model
 from tensorflow.keras.models import Sequential
 from tensorflow.keras.layers import Dense