wboler05/pso_neural_net

Change to Classification Output

Closed this issue · 2 comments

We should implement our outputs as full classification, based on the number of affected customers, instead of trying to predict the regression from a single node. This should make it easier for the ANN to train for a specific output. There will be 5 nodes, as follows:

  • A = 0
  • 0 < A <= 10
  • 10 < A <= 100
  • 100 < A <= 1000
  • 1000 < A

where each of the nodes represents the number of customers expected to be affected (severity).

We may need to balance the testing data, but we'll do so later if it becomes an issue. The last two levels may be difficult to train for.

This has been completed.