marinone94's picture
Training in progress, epoch 0
1ce325b
raw history blame
No virus
439 Bytes
#ifndef LM_BUILDER_DISCOUNT_H
#define LM_BUILDER_DISCOUNT_H
#include <algorithm>
#include <stdint.h>
namespace lm {
namespace builder {
struct Discount {
float amount[4];
float Get(uint64_t count) const {
return amount[std::min<uint64_t>(count, 3)];
}
float Apply(uint64_t count) const {
return static_cast<float>(count) - Get(count);
}
};
} // namespace builder
} // namespace lm
#endif // LM_BUILDER_DISCOUNT_H