GymIntentClassifier.java

/*
 * chatbot.component is added for Assignment 3 (Language Understanding)
 * 
 * WeatherIntentClassifier.java is added for Assignment 3 (Language
 * Understanding)
 */

package chatbot.component;

public class GymIntentClassifier {
	
	private static String[] intentDictionary;
	
	public GymIntentClassifier() {
		initializeIntentDictionary();
	}
	
	/**
	 * Create a dictionary of intents (under Weather domain)  
	 */
	private void initializeIntentDictionary() {
		intentDictionary = new String[]{"gym", "muscle", "exercise", "location", "time", "reps", "sets"};
		System.out.print("Intents: (");
		for(int i=0;i<intentDictionary.length;i++) {
			System.out.print(intentDictionary[i]);
			if(i!=intentDictionary.length-1) {
				System.out.print(", ");
			}
		}
		System.out.println(")");
	}
	
	/**
	 * Calculate the given meesage's score for each intent. The chatbot will
	 * select the intent with the *highest* score.
	 * 
	 * The initial score of each domain is 0.0.
	 * 
	 * @param nowInputText	An English message sent from the user.
	 * @return				An Double array that contains the score of each
	 * 						intent.
	 */
	private Double[] calculateIntentScores(String nowInputText) {
		
		//DO NOT change the following 4 lines
		//initiate all the scores to 0.0 
		Double[] scoreArray = new Double[intentDictionary.length];
		for(int i=0;i<scoreArray.length;i++) {
			scoreArray[i] = Double.valueOf(0.0);
		}
		
		//============= Please Modify Here (begins) =============== 
		
		//The following is the part you need to modify. 
		//This current version just assign random values to each intent.
		String newMessage = nowInputText.toLowerCase();
        for (int i = 0; i < intentDictionary.length; i++) {
            String intent = intentDictionary[i].toLowerCase();
            if (newMessage.contains(intent)) {
                scoreArray[i] = 1.00;
            }
        }
        /*
		String[] repsDictionary = new String[] {"reps"};
		for(String repsKeyword: repsDictionary) {
			if(nowInputText.toLowerCase().indexOf(repsKeyword)>=0) {
				scoreArray[1] = scoreArray[1].doubleValue()+1.0;
			}
		}
		
		String[] setsDictionary = new String[] {"sets"};
		for(String setsKeyword: setsDictionary) {
			if(nowInputText.toLowerCase().indexOf(setsKeyword)>=0) {
				scoreArray[2] = scoreArray[2].doubleValue()+1.0;
			}
		}
		
		String[] timeDictionary = new String[] {"time"};
		for(String timeKeyword: timeDictionary) {
			if(nowInputText.toLowerCase().indexOf(timeKeyword)>=0) {
				scoreArray[3] = scoreArray[3].doubleValue()+1.0;
			}
		}*/
		/*
		String message = nowInputText.toLowerCase();
		for (int i = 0; i < intentDictionary.length; i++) {
			String intent = intentDictionary[i].toLowerCase();
			if (message.contains(intent)) {
				scoreArray[i] = 1.00;
			}
	    }*/
		//============= Please Modify Here (ends) =============== 
		
		//do not change the following lines
		if(scoreArray.length!=intentDictionary.length) {
			System.err.println("The score array size does not equal to the intent array size.");
			System.exit(1);
		}
		for(Double nowValue: scoreArray) {
			if(nowValue==null) {
				System.err.println("The score array contains null values.");
				System.exit(1);
			}
		}
		return scoreArray;
	}
	
	/**
	 * Input:
	 * 	nowInputText: the message that the user sent to your chatbot
	 * 
	 * Output:
	 * 	the label (intent) name string
	 * 
	 * @param nowInputText	An English message sent from the user.
	 * @return 				The name of the intent.
	 * 
	 */
	public String getLabel(String nowInputText) {
		Double[] intentScores = calculateIntentScores(nowInputText);
		Double nowMaxScore = null;
		int nowMaxIndex = -1;
		System.out.print("Intent Scores: (");
		for(int i=0;i<intentScores.length;i++){
			System.out.print(intentScores[i].doubleValue());
			if(i!=intentScores.length-1) {
				System.out.print(", ");
			}
			if(nowMaxScore==null||nowMaxIndex==-1||intentScores[i].doubleValue()>nowMaxScore.doubleValue()) {
				nowMaxIndex = i;
				nowMaxScore = intentScores[i].doubleValue();
			}
		}
		System.out.println(")");
		return intentDictionary[nowMaxIndex];
	}

}