Sparse featurizer for unit presence. More...

Public Member Functions
	UnitPresenceFeaturizer ()

Public Member Functions inherited from cherrypi::UnitAttributeFeaturizer
virtual	~UnitAttributeFeaturizer ()=default

virtual Data	extract (State *state, UnitsInfo::Units const &units, Rect const &boundingBox=Rect())
	Extract unit features for a given set of units. More...

Data	extract (State *state, Rect const &boundingBox=Rect())
	Extract unit features for all live units. More...

Data	extract (State *state, UnitFilter filter, Rect const &boundingBox=Rect())
	Extract unit features for all live units that pass the given filter. More...

FeatureData	toSpatialFeature (Data const &data, SubsampleMethod pooling=SubsampleMethod::Sum) const
	Embeds the unit attribute data into a spatial feature. More...

void	toSpatialFeature (FeatureData *dest, Data const &data, SubsampleMethod pooling=SubsampleMethod::Sum) const
	Embeds the unit attribute data into a spatial feature. More...

Protected Member Functions
virtual void	extractUnit (TensorDest acc, Unit *) override
	Reimplement this in actual featurizers. More...

Additional Inherited Members
Public Types inherited from cherrypi::UnitAttributeFeaturizer
using	UnitFilter = std::function< bool(Unit *)>

using	TensorDest = torch::TensorAccessor< float, 1 >

Public Attributes inherited from cherrypi::UnitAttributeFeaturizer
std::shared_ptr< BaseJitter >	jitter = std::make_shared<NoJitter>()
	Optional jittering of unit positions. More...

CustomFeatureType	type

std::string	name

int	numChannels

Detailed Description

Sparse featurizer for unit presence.

This will produce a binary feature with a single channel: 0 if there is no unit, 1 if there is a unit.

Constructor & Destructor Documentation

cherrypi::UnitPresenceFeaturizer::UnitPresenceFeaturizer ( )

inline

virtual void cherrypi::UnitPresenceFeaturizer::extractUnit	(	TensorDest	acc,
		Unit *	unit
	)

inlineoverrideprotectedvirtual

Reimplement this in actual featurizers.

This function is expected to set acc[0], ..., acc[numChannels-1]

The documentation for this struct was generated from the following file: