Package org.apache.mahout.common

Examples of org.apache.mahout.common.FileLineIterator


   *          data path
   * @return
   */
  public static int[] extractLabels(Dataset dataset, FileSystem fs, Path path) throws IOException {
    FSDataInputStream input = fs.open(path);
    FileLineIterator iterator = new FileLineIterator(input);
   
    int[] labels = new int[dataset.nbInstances()];
    DataConverter converter = new DataConverter(dataset);
   
    int index = 0;
   
    while (iterator.hasNext()) {
      labels[index++] = converter.convert(0, iterator.next()).label;
    }
   
    iterator.close();
   
    return labels;
  }
View Full Code Here


    files = out.listFiles();
    assertEquals("files Size: " + files.length + " is not: " + words.length, files.length, words.length);
    for (File file : files) {
      //should only be one line in the file, and it should be label label
      FileLineIterator it = new FileLineIterator(file);
      String line = it.next().trim();
      assertFalse(it.hasNext());
      String label = "animal" + '\t' + file.getName();
      assertEquals(line + ":::: is not equal to " + label + "::::", line, label);
    }
  }
View Full Code Here

  }
 
  @Override
  protected DataModel buildModel() throws IOException {
    FastByIDMap<Collection<Preference>> data = new FastByIDMap<Collection<Preference>>();
    FileLineIterator iterator = new FileLineIterator(getDataFile(), false);
    processFile(iterator, data, false);
    return new GenericDataModel(GenericDataModel.toDataMap(data, true));
  }
View Full Code Here

   * <pre>
   * term DocFreq Index
   * </pre>
   */
  public static String[] loadTermDictionary(InputStream is) throws IOException {
    FileLineIterator it = new FileLineIterator(is);
   
    int numEntries = Integer.parseInt(it.next());
    // System.out.println(numEntries);
    String[] result = new String[numEntries];
   
    while (it.hasNext()) {
      String line = it.next();
      if (line.startsWith("#")) {
        continue;
      }
      String[] tokens = VectorHelper.TAB_PATTERN.split(line);
      if (tokens.length < 3) {
View Full Code Here

   
    StringBuilder content = new StringBuilder();
    content.append(header);
    NumberFormat decimalFormatter = new DecimalFormat("0000");
    File dumpFile = new File(dumpFilePath);
    FileLineIterator it;
    if (dumpFilePath.endsWith(".bz2")) {
      // default compression format from http://download.wikimedia.org
      CompressionCodec codec = new BZip2Codec();
      it = new FileLineIterator(codec.createInputStream(new FileInputStream(dumpFile)));
    } else {
      // assume the user has previously de-compressed the dump file
      it = new FileLineIterator(dumpFile);
    }
    int filenumber = 0;
    while (it.hasNext()) {
      String thisLine = it.next();
      if (thisLine.trim().startsWith("<page>")) {
        boolean end = false;
        while (thisLine.trim().startsWith("</page>") == false) {
          content.append(thisLine).append('\n');
          if (it.hasNext()) {
            thisLine = it.next();
          } else {
            end = true;
            break;
          }
        }
View Full Code Here

      try {
       
        averageDiffs.clear();
        allRecommendableItemIDs.clear();
       
        FileLineIterator iterator = new FileLineIterator(dataFile, false);
        String firstLine = iterator.peek();
        while ((firstLine.length() == 0) || (firstLine.charAt(0) == COMMENT_CHAR)) {
          iterator.next();
          firstLine = iterator.peek();
        }
        char delimiter = FileDataModel.determineDelimiter(firstLine, 2);
        long averageCount = 0L;
        while (iterator.hasNext()) {
          averageCount = processLine(iterator.next(), delimiter, averageCount);
        }
       
        pruneInconsequentialDiffs();
        updateAllRecommendableItems();
       
View Full Code Here

   
    this.dataFile = dataFile.getAbsoluteFile();
    this.lastModified = dataFile.lastModified();
    this.lastUpdateFileModified = readLastUpdateFileModified();
   
    FileLineIterator iterator = new FileLineIterator(dataFile, false);
    String firstLine = iterator.peek();
    while ((firstLine.length() == 0) || (firstLine.charAt(0) == COMMENT_CHAR)) {
      iterator.next();
      firstLine = iterator.peek();
    }
    iterator.close();
    delimiter = determineDelimiter(firstLine, 2);
    hasPrefValues = firstLine.indexOf(delimiter, firstLine.indexOf(delimiter) + 1) >= 0;
   
    this.reloadLock = new ReentrantLock();
    this.transpose = transpose;
View Full Code Here

    if (hasPrefValues) {
     
      if (loadFreshData) {
       
        FastByIDMap<Collection<Preference>> data = new FastByIDMap<Collection<Preference>>();
        FileLineIterator iterator = new FileLineIterator(dataFile, false);
        processFile(iterator, data, false);
       
        for (File updateFile : findUpdateFiles()) {
          processFile(new FileLineIterator(updateFile, false), data, false);
        }
       
        return new GenericDataModel(GenericDataModel.toDataMap(data, true));
       
      } else {
       
        FastByIDMap<PreferenceArray> rawData = ((GenericDataModel) delegate).getRawUserData();
       
        for (File updateFile : findUpdateFiles()) {
          processFile(new FileLineIterator(updateFile, false), rawData, true);
        }
       
        return new GenericDataModel(rawData);
       
      }
     
    } else {
     
      if (loadFreshData) {
       
        FastByIDMap<FastIDSet> data = new FastByIDMap<FastIDSet>();
        FileLineIterator iterator = new FileLineIterator(dataFile, false);
        processFileWithoutID(iterator, data);
       
        for (File updateFile : findUpdateFiles()) {
          processFileWithoutID(new FileLineIterator(updateFile, false), data);
        }
       
        return new GenericBooleanPrefDataModel(data);
       
      } else {
       
        FastByIDMap<FastIDSet> rawData = ((GenericBooleanPrefDataModel) delegate).getRawUserData();
       
        for (File updateFile : findUpdateFiles()) {
          processFileWithoutID(new FileLineIterator(updateFile, false), rawData);
        }
       
        return new GenericBooleanPrefDataModel(rawData);
       
      }
View Full Code Here

      try {

        averageDiffs.clear();
        allRecommendableItemIDs.clear();
       
        FileLineIterator iterator = new FileLineIterator(dataFile, false);
        String firstLine = iterator.peek();
        while (firstLine.length() == 0 || firstLine.charAt(0) == COMMENT_CHAR) {
          iterator.next();
          firstLine = iterator.peek();
        }
        char delimiter = FileDataModel.determineDelimiter(firstLine);
        long averageCount = 0L;
        while (iterator.hasNext()) {
          averageCount = processLine(iterator.next(), delimiter, averageCount);
        }
       
        pruneInconsequentialDiffs();
        updateAllRecommendableItems();
       
View Full Code Here

   * <pre>
   * term DocFreq Index
   * </pre>
   */
  private static String[] loadTermDictionary(InputStream is) throws IOException {
    FileLineIterator it = new FileLineIterator(is);
   
    int numEntries = Integer.parseInt(it.next());
    // System.out.println(numEntries);
    String[] result = new String[numEntries];
   
    while (it.hasNext()) {
      String line = it.next();
      if (line.startsWith("#")) {
        continue;
      }
      String[] tokens = VectorHelper.TAB_PATTERN.split(line);
      if (tokens.length < 3) {
View Full Code Here

TOP

Related Classes of org.apache.mahout.common.FileLineIterator

Copyright © 2018 www.massapicom. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.