public void generateUsers(String filename, int nextUserId, List<UserType> userTypes) {
dataGenerator.setNextUserId(nextUserId);
System.out.println("Generating users...");
List<User> allUsers = dataGenerator.generateUsers(userTypes);
System.out.println("Saving users into '" + filename + "'");
DataUtils.saveUsers(filename, allUsers);
}
public void create() {
System.out.println("Creating data for the credit worthiness (score) use case:");
System.out.println(" Number of users in the training set: " + nTrainingUsers);
System.out.println(" Number of users in the testing set: " + nTestUsers);
System.out.println("___________________________________________________________");
List<UserType> trainingUserTypes = createUserTypes(nTrainingUsers);
int userIdSequenceStart = 1;
generateUsers(TRAINING_USERS_FILENAME, userIdSequenceStart, trainingUserTypes);
dataGenerator.setNoiseOn(true);
List<UserType> testUserTypes = createUserTypes(nTestUsers);
userIdSequenceStart = 500000;
// generateUsers(TEST_USERS_FILENAME, 2*nTrainingUsers, testUserTypes);
generateUsers(TEST_USERS_FILENAME, userIdSequenceStart, testUserTypes);
System.out.println("Done!");
}
