c4dt / predictive-maintenance

An experiment to decentralize learning for predictive maintenance
0 stars 0 forks source link

Train using anomalous data [MLP / LR] #11

Closed lanterno closed 1 year ago

lanterno commented 1 year ago

In two previous attempts, we've tried transfer learning to MLP and to LR models. However, both didn't have good results.

In this issue, We want to modify the training data to add some anomalous images as well.

Once that's done, we review the results.

To review the results of the two experiments, check these two notebooks:

https://github.com/c4dt/predictive-maintenance/blob/main/experiments/linear_regression.ipynb https://github.com/c4dt/predictive-maintenance/blob/main/experiments/mlp.ipynb