Homec4science
Diffusion LPBF multi-material composition monitoring (master) Browse

R12999/Contrastive Learners/Circle loss LPBF/Classifiers/Logistic_regression.py
6cb20207fb21master

Last Change
Raw File

Logistic_regression.py
View Options

"""
@author: srpv
contact: vigneashwara.solairajapandiyan@empa.ch, vigneashpandiyan@gmail.com
The codes in this following script will be used for the publication of the following work
"Qualify-As-You-Go: Sensor Fusion of Optical and Acoustic Signatures with Contrastive Deep Learning for Multi-Material Composition Monitoring in Laser Powder Bed Fusion Process"
@any reuse of this code should be authorized by the first owner, code author
"""
#%% Libraries required:
from sklearn.metrics import classification_report, confusion_matrix
from sklearn.metrics import confusion_matrix, ConfusionMatrixDisplay
from sklearn.model_selection import RepeatedStratifiedKFold
import joblib
from sklearn.model_selection import cross_val_score
import matplotlib.pyplot as plt
import numpy as np
from sklearn import metrics
import pandas as pd
from sklearn.linear_model import LogisticRegression
import os
from sklearn.model_selection import RepeatedStratifiedKFold
from sklearn.model_selection import cross_val_score
from numpy import mean
from numpy import std
# %%
def LR(X_train, X_test, y_train, y_test, classes, total_path):
"""
Trains a Logistic Regression model, evaluates its performance, and saves the model.
Parameters:
X_train (array-like): Training data features.
X_test (array-like): Test data features.
y_train (array-like): Training data labels.
y_test (array-like): Test data labels.
classes (array-like): List of class labels.
total_path (str): Path to save the model and evaluation graphs.
Returns:
None
"""
print('Model to be trained is LR')
model = LogisticRegression(max_iter=1000, random_state=123)
model.fit(X_train, y_train)
cv = RepeatedStratifiedKFold(n_splits=10, n_repeats=3, random_state=1)
scores = cross_val_score(model, X_test, y_test, scoring='accuracy', cv=cv, n_jobs=-1)
print('Accuracy: %.3f (%.3f)' % (mean(scores), std(scores)))
predictions = model.predict(X_test)
print("LogisticRegression Accuracy:", metrics.accuracy_score(y_test, predictions))
print(classification_report(y_test, predictions))
print(confusion_matrix(y_test, predictions))
graph_name1 = 'LR'+'_without normalization w/o Opt'
graph_name2 = 'Logistic Regression'
graph_1 = 'LR'+'_Confusion_Matrix'+'_'+'No_Opt'+'.png'
graph_2 = 'LR'+'_Confusion_Matrix'+'_'+'Opt'+'.png'
titles_options = [(graph_name1, None, graph_1),
(graph_name2, 'true', graph_2)]
for title, normalize, graphname in titles_options:
plt.figure(figsize=(20, 10), dpi=400)
disp = ConfusionMatrixDisplay.from_estimator(model, X_test, y_test,
display_labels=classes,
cmap=plt.cm.RdPu, xticks_rotation='vertical',
normalize=normalize, values_format='0.2f')
plt.title(title, size=12)
plt.savefig(os.path.join(total_path, graphname), bbox_inches='tight', dpi=400)
plt.show()
savemodel = 'LR'+'_model'+'.sav'
joblib.dump(model, savemodel)
c4science · Help