Workbench: Wrong training set preselected

[PetrToman]

If *_eval.egb file is present, it is preselected in the training dialog. It would be more useful to preselect a training file (using *_train.egb mask).

[PetrToman]

Here is a solution:

• Add this to EncogWorkBench class:

import java.util.Collections;
import java.util.Comparator;
import org.encog.workbench.util.FileUtil;

    public List<ProjectTraining> getTrainingData(final String sortFirst) {
        List<ProjectTraining> list = getTrainingData();

        Collections.sort(list, new Comparator<ProjectTraining>() {
            @Override
            public int compare(ProjectTraining p1, ProjectTraining p2) {
                boolean isFirst1 = FileUtil.getFileName(p1.getFile()).endsWith(sortFirst);
                boolean isFirst2 = FileUtil.getFileName(p2.getFile()).endsWith(sortFirst);

                if (isFirst1 && !isFirst2) return -1;
                if (isFirst2 && !isFirst1) return 1;

                return p1.getName().compareTo(p2.getName());
            }
        });

        return list;
    }

• In NetworkAndTrainingDialog change line 91 to:

this.trainingSets = EncogWorkBench.getInstance().getTrainingData("_train");

• In TrainDialog change line 58 to:

List<ProjectTraining> list = EncogWorkBench.getInstance().getTrainingData("_eval");

• Add this to TrainDialog:

import org.encog.workbench.util.FileUtil;

    private void ensureNonEvalUnselection()
    {
        JComboBox comboValidation = (JComboBox) this.comboValidation.getField();
        ProjectTraining selectedItem = (ProjectTraining) comboValidation.getSelectedItem();

        if (selectedItem == null || !FileUtil.getFileName(selectedItem.getFile()).endsWith("_eval"))
        {
            comboValidation.setSelectedIndex(-1);
        }
    }

• In TrainDialog change line 55:

((JComboBox)this.comboValidation.getField()).setSelectedIndex(-1);

to:

ensureNonEvalUnselection();

[jeffheaton]

Good suggestion, added the suggested code.