All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • org.jpmml.converter.CategoryManager Maven / Gradle / Ivy

    There is a newer version: 1.5.10
    Show newest version
    /*
 * Copyright (c) 2018 Villu Ruusmann
 *
 * This file is part of JPMML-Converter
 *
 * JPMML-Converter is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * JPMML-Converter is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with JPMML-Converter.  If not, see .
 */
package org.jpmml.converter;

import java.util.Collection;
import java.util.LinkedHashMap;
import java.util.LinkedHashSet;
import java.util.Map;
import java.util.Set;
import java.util.function.Predicate;

import org.dmg.pmml.FieldName;

public class CategoryManager extends ValueManager> {

	public CategoryManager(){
	}

	public CategoryManager(Map> valueMap){
		super(valueMap);
	}

	public CategoryManager fork(FieldName name, Collection values){
		return fork(name, new LinkedHashSet<>(values));
	}

	@Override
	public CategoryManager fork(FieldName name, Set values){
		Map> valueMap = new LinkedHashMap<>(getValueMap());

		valueMap.put(name, values);

		return new CategoryManager(valueMap);
	}

	public Predicate getValueFilter(FieldName name){
		Set values = getValue(name);

		Predicate predicate = new Predicate(){

			@Override
			public boolean test(Object value){

				if(values != null){
					return values.contains(value);
				}

				return true;
			}
		};

		return predicate;
	}
}    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy