Sha256: d66e9a0cb1bee9c1e74c25d738ce1ddba927380f98cc22da81eab4486f280c12
Contents?: true
Size: 1.68 KB
Versions: 11
Compression:
Stored size: 1.68 KB
Contents
/**
* Copyright (c) 2016-present, Facebook, Inc.
* All rights reserved.
*
* This source code is licensed under the MIT license found in the
* LICENSE file in the root directory of this source tree.
*/
#pragma once
#include <memory>
#include <random>
#include <utility>
#include <vector>
#include "matrix.h"
#include "real.h"
#include "utils.h"
#include "vector.h"
namespace fasttext {
class Loss;
class Model {
protected:
std::shared_ptr<Matrix> wi_;
std::shared_ptr<Matrix> wo_;
std::shared_ptr<Loss> loss_;
bool normalizeGradient_;
public:
Model(
std::shared_ptr<Matrix> wi,
std::shared_ptr<Matrix> wo,
std::shared_ptr<Loss> loss,
bool normalizeGradient);
Model(const Model& model) = delete;
Model(Model&& model) = delete;
Model& operator=(const Model& other) = delete;
Model& operator=(Model&& other) = delete;
class State {
private:
real lossValue_;
int64_t nexamples_;
public:
Vector hidden;
Vector output;
Vector grad;
std::minstd_rand rng;
State(int32_t hiddenSize, int32_t outputSize, int32_t seed);
real getLoss() const;
void incrementNExamples(real loss);
};
void predict(
const std::vector<int32_t>& input,
int32_t k,
real threshold,
Predictions& heap,
State& state) const;
void update(
const std::vector<int32_t>& input,
const std::vector<int32_t>& targets,
int32_t targetIndex,
real lr,
State& state);
void computeHidden(const std::vector<int32_t>& input, State& state) const;
real std_log(real) const;
static const int32_t kUnlimitedPredictions = -1;
static const int32_t kAllLabelsAsTarget = -1;
};
} // namespace fasttext
Version data entries
11 entries across 11 versions & 1 rubygems