|
Algorithm 1: WCIGDG |
| # Adversarial Training stage |
|
Input: source domain dataset , multiple auxiliary domain datasets
|
| 1: for epoch = 1 to epochs = N
do
|
| 2: Forward propagation to generate . |
| 3: Calculate and
|
| 4: Backward propagation to update and
|
| 5: Calculate
|
| 6: Backward propagation to update
|
| Return: Trained generator
|
| # Fault Diagnosis Model Training stage |
| Input: source domain dataset . |
| 1: Forward propagation to generate . |
| 2: for epoch = 1 to epochs = N do
|
| 3: Calculate based on the classification results of generated data and real data from source domain . |
| 4: Backward propagation to update
|
| Return: Trained classifier
|