PClassifierInfo.h

Go to the documentation of this file.

/*
 * Copyright (C) 2015-2019 Digital Barriers plc. All rights reserved.
 * Contact: http://www.digitalbarriers.com/
 *
 * This file is part of the Papillon SDK.
 *
 * You can't use, modify or distribute any part of this file without
 * the explicit written agreements of Digital Barriers.
 */

#pragma once
#include <PObject.h>
#include <PString.h>

PAPILLON_BEGIN_NAMESPACE

class PStringList;

class PAPILLON_API_CORE PClassifierInfo : public PObject {
 public:
    PClassifierInfo();

    explicit PClassifierInfo(const PStringList& classLabels, float threshold = 0, const PString& unknownLabel = "unknown");

    explicit PClassifierInfo(const PClassifierInfo& other);

    ~PClassifierInfo() override;

    PClassifierInfo& operator=(const PClassifierInfo& other);

    int32 GetNumClasses() const;

    const PStringList& GetClassLabels() const;

    float GetThreshold() const;

    const PString& GetClassLabel(int32 classIndex) const;
};

PAPILLON_END_NAMESPACE