redux-scraper/sorbet/rbi/gems/libmf@0.4.0.rbi

# typed: true

# DO NOT EDIT MANUALLY
# This is an autogenerated file for types exported from the `libmf` gem.
# Please instead update this file by running `bin/tapioca gem libmf`.


# source://libmf//lib/libmf/matrix.rb#1
module Libmf
  class << self
    # Returns the value of attribute ffi_lib.
    #
    # source://libmf//lib/libmf.rb#13
    def ffi_lib; end

    # Sets the attribute ffi_lib
    #
    # @param value the value to set the attribute ffi_lib to.
    #
    # source://libmf//lib/libmf.rb#13
    def ffi_lib=(_arg0); end
  end
end

# source://libmf//lib/libmf.rb#10
class Libmf::Error < ::StandardError; end

# source://libmf//lib/libmf/ffi.rb#2
module Libmf::FFI
  extend ::FFI::Library

  def calc_accuracy(*_arg0); end
  def calc_auc(*_arg0); end
  def calc_gkl(*_arg0); end
  def calc_logloss(*_arg0); end
  def calc_mae(*_arg0); end
  def calc_mpr(*_arg0); end
  def calc_rmse(*_arg0); end
  def mf_cross_validation(*_arg0); end
  def mf_destroy_model(*_arg0); end
  def mf_get_default_param(*_arg0); end
  def mf_load_model(*_arg0); end
  def mf_predict(*_arg0); end
  def mf_read_problem(*_arg0); end
  def mf_save_model(*_arg0); end
  def mf_train(*_arg0); end
  def mf_train_with_validation(*_arg0); end

  class << self
    def calc_accuracy(*_arg0); end
    def calc_auc(*_arg0); end
    def calc_gkl(*_arg0); end
    def calc_logloss(*_arg0); end
    def calc_mae(*_arg0); end
    def calc_mpr(*_arg0); end
    def calc_rmse(*_arg0); end
    def mf_cross_validation(*_arg0); end
    def mf_destroy_model(*_arg0); end
    def mf_get_default_param(*_arg0); end
    def mf_load_model(*_arg0); end
    def mf_predict(*_arg0); end
    def mf_read_problem(*_arg0); end
    def mf_save_model(*_arg0); end
    def mf_train(*_arg0); end
    def mf_train_with_validation(*_arg0); end
  end
end

# source://libmf//lib/libmf/ffi.rb#38
class Libmf::FFI::Model < ::FFI::Struct
  class << self
    # source://libmf//lib/libmf/ffi.rb#47
    def release(pointer); end
  end
end

# source://libmf//lib/libmf/ffi.rb#7
class Libmf::FFI::Node < ::FFI::Struct; end

# source://libmf//lib/libmf/ffi.rb#20
class Libmf::FFI::Parameter < ::FFI::Struct; end

# source://libmf//lib/libmf/ffi.rb#13
class Libmf::FFI::Problem < ::FFI::Struct; end

# source://libmf//lib/libmf/matrix.rb#2
class Libmf::Matrix
  # @return [Matrix] a new instance of Matrix
  #
  # source://libmf//lib/libmf/matrix.rb#5
  def initialize; end

  # Returns the value of attribute data.
  #
  # source://libmf//lib/libmf/matrix.rb#3
  def data; end

  # source://libmf//lib/libmf/matrix.rb#9
  def push(row_index, column_index, value); end
end

# source://libmf//lib/libmf/model.rb#2
class Libmf::Model
  # @return [Model] a new instance of Model
  #
  # source://libmf//lib/libmf/model.rb#3
  def initialize(**options); end

  # source://libmf//lib/libmf/model.rb#91
  def accuracy(data); end

  # source://libmf//lib/libmf/model.rb#99
  def auc(data, transpose); end

  # source://libmf//lib/libmf/model.rb#63
  def bias; end

  # source://libmf//lib/libmf/model.rb#55
  def columns; end

  # @raise [Error]
  #
  # source://libmf//lib/libmf/model.rb#26
  def cv(data, folds: T.unsafe(nil)); end

  # source://libmf//lib/libmf/model.rb#59
  def factors; end

  # @raise [Error]
  #
  # source://libmf//lib/libmf/model.rb#7
  def fit(data, eval_set: T.unsafe(nil)); end

  # source://libmf//lib/libmf/model.rb#83
  def gkl(data); end

  # @raise [Error]
  #
  # source://libmf//lib/libmf/model.rb#46
  def load_model(path); end

  # source://libmf//lib/libmf/model.rb#87
  def logloss(data); end

  # source://libmf//lib/libmf/model.rb#79
  def mae(data); end

  # source://libmf//lib/libmf/model.rb#95
  def mpr(data, transpose); end

  # source://libmf//lib/libmf/model.rb#67
  def p_factors(format: T.unsafe(nil)); end

  # source://libmf//lib/libmf/model.rb#22
  def predict(row, column); end

  # source://libmf//lib/libmf/model.rb#71
  def q_factors(format: T.unsafe(nil)); end

  # source://libmf//lib/libmf/model.rb#75
  def rmse(data); end

  # source://libmf//lib/libmf/model.rb#51
  def rows; end

  # @raise [Error]
  #
  # source://libmf//lib/libmf/model.rb#34
  def save(path); end

  # @raise [Error]
  #
  # source://libmf//lib/libmf/model.rb#34
  def save_model(path); end

  private

  # source://libmf//lib/libmf/model.rb#105
  def _factors(ptr, n, format); end

  # @raise [Error]
  #
  # source://libmf//lib/libmf/model.rb#161
  def create_problem(data); end

  # @raise [Error]
  #
  # source://libmf//lib/libmf/model.rb#116
  def model; end

  # source://libmf//lib/libmf/model.rb#121
  def param; end

  class << self
    # source://libmf//lib/libmf/model.rb#40
    def load(path); end
  end
end

# source://libmf//lib/libmf/version.rb#2
Libmf::VERSION = T.let(T.unsafe(nil), String)