Commit ee41cdc

committed

more docs

1 parent e45db84 commit ee41cdcCopy full SHA for ee41cdc

File tree

+10

-1

lines changed

+10

-1

lines changed

Lines changed: 3 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -19,4 +19,6 @@ docs/_build`
`19`	`19`	`.vscode`
`20`	`20`	`app/__pycache__`
`21`	`21`	`app/data/__pycache__`
`22`		`-app/data/data_banknote_authentication.txt`
	`22`	`+app/data/data_banknote_authentication.txt`
	`23`	`+LICENSE`
	`24`	`+README.md`

Lines changed: 6 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -15,21 +15,27 @@ def get_predictions(data):`
`15`	`15`	`:param data: Dict with keys representing features and values representing the associated value`
`16`	`16`	`:return: Dict with keys 'predicted_class' (class predicted) and 'predicted_prob' (probability of prediction)`
`17`	`17`	`"""`
	`18`	`+ # Convert new data dict as a DataFrame and reshape the columns to suit the model`
`18`	`19`	`new_data = {k: [v] for k, v in data.items()}`
`19`	`20`
`20`	`21`	`new_data_df = pd.DataFrame.from_dict(new_data)`
	`22`	`+`
`21`	`23`	`new_data_df = new_data_df[['variance_of_wavelet', 'skewness_of_wavelet',`
`22`	`24`	`'curtosis_of_wavelet', 'entropy_of_wavelet']]`
`23`	`25`
	`26`	`+ # Load saved standardising object`
`24`	`27`	`scaler = joblib.load('app/data/scaler.joblib')`
`25`	`28`	`logger.debug('Saved standardising object successfully loaded')`
`26`	`29`
	`30`	`+ # Load saved keras model`
`27`	`31`	`model = tf.keras.models.load_model('app/data/banknote_authentication_model.h5')`
`28`	`32`	`logger.debug('Saved ANN model loaded successfully')`
`29`	`33`
	`34`	`+ # Scale new data using the loaded object`
`30`	`35`	`X = scaler.transform(new_data_df.values)`
`31`	`36`	`logger.debug('Incoming data successfully standardised with saved object')`
`32`	`37`
	`38`	`+ # Make new predictions from the newly scaled data and return this prediction`
`33`	`39`	`preds = model.predict(X)`
`34`	`40`
`35`	`41`	`return preds`

Lines changed: 1 addition & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -62,6 +62,7 @@ def get_prediction(incoming_data: Features):`
`62`	`62`
`63`	`63`
`64`	`64`	`if __name__ == "__main__":`
	`65`	`+ # Run app with uvicorn with port and host specified. Host needed for docker port mapping`
`65`	`66`	`uvicorn.run(app, port=8000, host="0.0.0.0")`
`66`	`67`
`67`	`68`

Comments

(0)