Examples with LabelList de.lmu.ifi.dbs.elki.data.LabelList used on opensource projects

Example 1 with LabelList

use of de.lmu.ifi.dbs.elki.data.LabelList in project elki by elki-project.

the class ExternalIDFilter method filter.

@Override
public MultipleObjectsBundle filter(MultipleObjectsBundle objects) {
    MultipleObjectsBundle bundle = new MultipleObjectsBundle();
    // Find a labellist column
    boolean done = false;
    boolean keeplabelcol = false;
    for (int i = 0; i < objects.metaLength(); i++) {
        SimpleTypeInformation<?> meta = objects.meta(i);
        // Skip non-labellist columns - or if we already had a labellist
        if (done || !LabelList.class.equals(meta.getRestrictionClass())) {
            bundle.appendColumn(meta, objects.getColumn(i));
            continue;
        }
        done = true;
        // We split the label column into two parts
        List<ExternalID> eidcol = new ArrayList<>(objects.dataLength());
        List<LabelList> lblcol = new ArrayList<>(objects.dataLength());
        // Split the column
        ArrayList<String> lbuf = new ArrayList<>();
        for (Object obj : objects.getColumn(i)) {
            if (obj != null) {
                LabelList ll = (LabelList) obj;
                int off = externalIdIndex >= 0 ? externalIdIndex : (ll.size() - externalIdIndex);
                eidcol.add(new ExternalID(ll.get(off)));
                lbuf.clear();
                for (int j = 0; j < ll.size(); j++) {
                    if (j == off) {
                        continue;
                    }
                    lbuf.add(ll.get(j));
                }
                lblcol.add(LabelList.make(lbuf));
                if (ll.size() > 0) {
                    keeplabelcol = true;
                }
            } else {
                eidcol.add(null);
                lblcol.add(null);
            }
        }
        bundle.appendColumn(TypeUtil.EXTERNALID, eidcol);
        // Only add the label column when it's not empty.
        if (keeplabelcol) {
            bundle.appendColumn(meta, lblcol);
        }
    }
    return bundle;
}

Example 2 with LabelList

use of de.lmu.ifi.dbs.elki.data.LabelList in project elki by elki-project.

the class ByLabelFilterTest method parameters.

/**
 * Test with parameter s as the label to look for.
 */
@Test
public void parameters() {
    String s = "yes";
    String filename = UNITTEST + "label-selection-test-1.csv";
    ByLabelFilter filter = // 
    new ELKIBuilder<>(ByLabelFilter.class).with(ByLabelFilter.Parameterizer.LABELFILTER_PATTERN_ID, // 
    s).build();
    MultipleObjectsBundle filteredBundle = readBundle(filename, filter);
    // Load the test data again without a filter.
    MultipleObjectsBundle unfilteredBundle = readBundle(filename);
    // Ensure the first column are the vectors.
    assertEquals("Dimensionality", getFieldDimensionality(unfilteredBundle, 0, TypeUtil.NUMBER_VECTOR_FIELD), getFieldDimensionality(filteredBundle, 0, TypeUtil.NUMBER_VECTOR_FIELD));
    // Verify that the filter selected all vectors which match the pattern.
    int count_match = 0;
    for (int row = 0; row < unfilteredBundle.dataLength(); row++) {
        LabelList ll = get(unfilteredBundle, row, 1, LabelList.class);
        if (ll.get(0).equals(s)) {
            count_match++;
        }
    }
    assertTrue("Expected at least one match", count_match > 0);
    assertEquals("Unexpected number of matches", count_match, filteredBundle.dataLength());
}

Example 3 with LabelList

use of de.lmu.ifi.dbs.elki.data.LabelList in project elki by elki-project.

the class SortByLabelFilterTest method defaultParameters.

/**
 * Test with default parameters.
 */
@Test
public void defaultParameters() {
    String filename = UNITTEST + "label-selection-test-1.csv";
    SortByLabelFilter filter = new ELKIBuilder<>(SortByLabelFilter.class).build();
    MultipleObjectsBundle bundle = readBundle(filename, filter);
    // Expect vectors to come first, labels second.
    getFieldDimensionality(bundle, 0, TypeUtil.NUMBER_VECTOR_FIELD);
    // Verify that the vectors are in alphabetical order.
    for (int row = 0; row < bundle.dataLength() - 1; row++) {
        LabelList llFirst = get(bundle, row, 1, LabelList.class);
        LabelList llSecond = get(bundle, row + 1, 1, LabelList.class);
        assertTrue("Expected alphabetical order", llFirst.get(0).compareToIgnoreCase(llSecond.get(0)) <= 0);
    }
}

Example 4 with LabelList

use of de.lmu.ifi.dbs.elki.data.LabelList in project elki by elki-project.

the class EvaluateRetrievalPerformance method match.

/**
 * Test whether two relation agree.
 *
 * @param ref Reference object
 * @param test Test object
 * @return {@code true} if the objects match
 */
protected static boolean match(Object ref, Object test) {
    if (ref == null) {
        return false;
    }
    // Cheap and fast, may hold for class labels!
    if (ref == test) {
        return true;
    }
    if (ref instanceof LabelList && test instanceof LabelList) {
        final LabelList lref = (LabelList) ref;
        final LabelList ltest = (LabelList) test;
        final int s1 = lref.size(), s2 = ltest.size();
        if (s1 == 0 || s2 == 0) {
            return false;
        }
        for (int i = 0; i < s1; i++) {
            String l1 = lref.get(i);
            if (l1 == null) {
                continue;
            }
            for (int j = 0; j < s2; j++) {
                if (l1.equals(ltest.get(j))) {
                    return true;
                }
            }
        }
    }
    // Fallback to equality, e.g. on class labels
    return ref.equals(test);
}

Example 5 with LabelList

use of de.lmu.ifi.dbs.elki.data.LabelList in project elki by elki-project.

the class AveragePrecisionAtK method match.

/**
 * Test whether two relation agree.
 *
 * @param ref Reference object
 * @param test Test object
 * @return {@code true} if the objects match
 */
protected static boolean match(Object ref, Object test) {
    if (ref == null) {
        return false;
    }
    // Cheap and fast, may hold for class labels!
    if (ref == test) {
        return true;
    }
    if (ref instanceof LabelList && test instanceof LabelList) {
        final LabelList lref = (LabelList) ref;
        final LabelList ltest = (LabelList) test;
        final int s1 = lref.size(), s2 = ltest.size();
        if (s1 == 0 || s2 == 0) {
            return false;
        }
        for (int i = 0; i < s1; i++) {
            String l1 = lref.get(i);
            if (l1 == null) {
                continue;
            }
            for (int j = 0; j < s2; j++) {
                if (l1.equals(ltest.get(j))) {
                    return true;
                }
            }
        }
    }
    // Fallback to equality, e.g. on class labels
    return ref.equals(test);
}