#include <toolkits/pattern_mining/fp_growth.hpp>
|
void | validate (const gl_sframe &data, const std::string &item, const std::vector< std::string > &features) const |
|
void | set_features (const std::vector< std::string > &_features) |
|
void | set_item (const std::string &_item) |
|
void | train (const gl_sframe &data) |
|
gl_sframe | extract_features (const gl_sframe &data) const |
|
gl_sframe | predict_topk (const gl_sframe &data, const std::string &score_function="confidence", const size_t &k=5) const |
|
size_t | get_version () const override |
|
void | save_impl (oarchive &oarc) const override |
|
void | load_version (iarchive &iarc, size_t version) override |
|
void | init_options (const std::map< std::string, flexible_type > &_options) override |
|
gl_sframe | get_frequent_patterns () const |
|
std::vector< std::string > | list_fields () |
|
const variant_type & | get_value_from_state (std::string key) |
|
const std::map< std::string, flexible_type > & | get_current_options () const |
|
std::map< std::string, flexible_type > | get_default_options () const |
|
const flexible_type & | get_option_value (const std::string &name) const |
|
const std::map< std::string, variant_type > & | get_state () const |
|
bool | is_trained () const |
|
void | set_options (const std::map< std::string, flexible_type > &_options) |
|
void | add_or_update_state (const std::map< std::string, variant_type > &dict) |
|
const std::vector< option_handling::option_info > & | get_option_info () const |
|
virtual const char * | name ()=0 |
|
virtual const std::string & | uid ()=0 |
|
void | save_to_url (const std::string &url, const variant_map_type &side_data={}) |
|
void | save_model_to_data (std::ostream &out) |
|
const std::map< std::string, std::vector< std::string > > & | list_functions () |
|
const std::vector< std::string > & | list_get_properties () |
|
const std::vector< std::string > & | list_set_properties () |
|
variant_type | call_function (const std::string &function, variant_map_type argument) |
|
variant_type | get_property (const std::string &property) |
|
variant_type | set_property (const std::string &property, variant_map_type argument) |
|
const std::string & | get_docstring (const std::string &symbol) |
|
virtual void | perform_registration () |
|
TODO: Add some comments hare about what the base class is supposed to be.
Definition at line 36 of file fp_growth.hpp.
◆ add_or_update_state()
void turi::ml_model_base::add_or_update_state |
( |
const std::map< std::string, variant_type > & |
dict | ) |
|
|
inherited |
Append the key value store of the model.
- Parameters
-
[in] | dict | Options (Key-Value pairs) to set |
◆ call_function()
variant_type turi::model_base::call_function |
( |
const std::string & |
function, |
|
|
variant_map_type |
argument |
|
) |
| |
|
inherited |
Calls a user defined function.
◆ extract_features()
gl_sframe turi::pattern_mining::fp_growth::extract_features |
( |
const gl_sframe & |
data | ) |
const |
Extract features from the training data.
- Parameters
-
[in] | Data | in the same format as the input data. |
◆ get_current_options()
const std::map<std::string, flexible_type>& turi::ml_model_base::get_current_options |
( |
| ) |
const |
|
inherited |
Get current options.
- Returns
- Dictionary containing current options.
Python side interface
Interfaces with the get_current_options function in the Python side.
◆ get_default_options()
std::map<std::string, flexible_type> turi::ml_model_base::get_default_options |
( |
| ) |
const |
|
inherited |
Get default options.
- Returns
- Dictionary with default options.
Python side interface
Interfaces with the get_default_options function in the Python side.
◆ get_docstring()
const std::string& turi::model_base::get_docstring |
( |
const std::string & |
symbol | ) |
|
|
inherited |
Returns the toolkit documentation for a function or property.
◆ get_frequent_patterns()
gl_sframe turi::pattern_mining::fp_growth::get_frequent_patterns |
( |
| ) |
const |
Get frequent itemsets using the FP-Growth algorithm.
◆ get_option_info()
Returns the option information struct for each of the set parameters.
◆ get_option_value()
const flexible_type& turi::ml_model_base::get_option_value |
( |
const std::string & |
name | ) |
const |
|
inherited |
Returns the value of an option. Throws an error if the option does not exist.
- Parameters
-
[in] | name | Name of the option to get. |
◆ get_property()
variant_type turi::model_base::get_property |
( |
const std::string & |
property | ) |
|
|
inherited |
◆ get_state()
const std::map<std::string, variant_type>& turi::ml_model_base::get_state |
( |
| ) |
const |
|
inherited |
Get model.
- Returns
- Model map.
◆ get_value_from_state()
const variant_type& turi::ml_model_base::get_value_from_state |
( |
std::string |
key | ) |
|
|
inherited |
Returns the value of a particular key from the state.
- Returns
- Value of a key model_base for details.
Python side interface
From the python side, this is interfaced with the get() function or the [] operator in python.
◆ get_version()
size_t turi::pattern_mining::fp_growth::get_version |
( |
| ) |
const |
|
overridevirtual |
◆ init_options()
void turi::pattern_mining::fp_growth::init_options |
( |
const std::map< std::string, flexible_type > & |
_options | ) |
|
|
overridevirtual |
Set one of the options in the algorithm. Use the option manager to set these options. If the option does not satisfy the conditions that the option manager has imposed on it. Errors will be thrown.
- Parameters
-
[in] | options | Options to set |
Reimplemented from turi::ml_model_base.
◆ is_trained()
bool turi::ml_model_base::is_trained |
( |
| ) |
const |
|
inherited |
Is this model trained.
- Returns
- True if already trained.
◆ list_fields()
std::vector<std::string> turi::ml_model_base::list_fields |
( |
| ) |
|
|
inherited |
Methods with already meaningful default implementations.
Lists all the keys accessible in the "model" map.
- Returns
- List of keys in the model map. model_base for details.
Python side interface
This is the function that the list_fields should call in python.
◆ list_functions()
const std::map<std::string, std::vector<std::string> >& turi::model_base::list_functions |
( |
| ) |
|
|
inherited |
Lists all the registered functions. Returns a map of function name to array of argument names for the function.
◆ list_get_properties()
const std::vector<std::string>& turi::model_base::list_get_properties |
( |
| ) |
|
|
inherited |
Lists all the get-table properties of the class.
◆ list_set_properties()
const std::vector<std::string>& turi::model_base::list_set_properties |
( |
| ) |
|
|
inherited |
Lists all the set-table properties of the class.
◆ load_version()
void turi::pattern_mining::fp_growth::load_version |
( |
iarchive & |
iarc, |
|
|
size_t |
version |
|
) |
| |
|
overridevirtual |
Loads a model previously saved at a particular version number. Should raise an exception on failure.
Reimplemented from turi::model_base.
◆ name()
virtual const char* turi::model_base::name |
( |
| ) |
|
|
pure virtualinherited |
Returns the name of the toolkit class, as exposed to client code. For example, the Python proxy for this instance will have a type with this name.
Note: this function is typically overridden using the BEGIN_CLASS_MEMBER_REGISTRATION macro.
◆ perform_registration()
virtual void turi::model_base::perform_registration |
( |
| ) |
|
|
virtualinherited |
Declare the base registration function. This class has to be handled specially; the macros don't work here due to the override declarations.
Reimplemented in turi::model_proxy.
◆ predict_topk()
gl_sframe turi::pattern_mining::fp_growth::predict_topk |
( |
const gl_sframe & |
data, |
|
|
const std::string & |
score_function = "confidence" , |
|
|
const size_t & |
k = 5 |
|
) |
| const |
Predict top-k rules
- Parameters
-
[in] | Data | in the same format as the input data. |
[in] | output_type | The scoring function used to score the results. |
[in] | k | The topk rules to predict. |
◆ register_defaults()
void turi::model_base::register_defaults |
( |
const std::string & |
fnname, |
|
|
const variant_map_type & |
arguments |
|
) |
| |
|
protectedinherited |
Registers default argument values
◆ register_docstring()
void turi::model_base::register_docstring |
( |
const std::pair< std::string, std::string > & |
fnname_docstring | ) |
|
|
protectedinherited |
Adds a docstring for the specified function or property name.
◆ register_function()
void turi::model_base::register_function |
( |
std::string |
fnname, |
|
|
const std::vector< std::string > & |
arguments, |
|
|
impl_fn |
fn |
|
) |
| |
|
protectedinherited |
Adds a function with the specified name, and argument list.
◆ register_getter()
void turi::model_base::register_getter |
( |
const std::string & |
propname, |
|
|
impl_fn |
getfn |
|
) |
| |
|
protectedinherited |
Adds a property getter with the specified name.
◆ register_setter()
void turi::model_base::register_setter |
( |
const std::string & |
propname, |
|
|
impl_fn |
setfn |
|
) |
| |
|
protectedinherited |
Adds a property setter with the specified name.
◆ save_impl()
void turi::pattern_mining::fp_growth::save_impl |
( |
oarchive & |
oarc | ) |
const |
|
overridevirtual |
◆ save_model_to_data()
void turi::model_base::save_model_to_data |
( |
std::ostream & |
out | ) |
|
|
inherited |
Save a toolkit class to a data stream.
◆ save_to_url()
void turi::model_base::save_to_url |
( |
const std::string & |
url, |
|
|
const variant_map_type & |
side_data = {} |
|
) |
| |
|
inherited |
Save a toolkit class to disk.
- Parameters
-
url | The destination url to store the class. |
sidedata | Any additional side information |
◆ set_features()
void turi::pattern_mining::fp_growth::set_features |
( |
const std::vector< std::string > & |
_features | ) |
|
|
inline |
Set the feature column names.
- Parameters
-
[in] | _features | Names of the "features" column. |
Definition at line 66 of file fp_growth.hpp.
◆ set_item()
void turi::pattern_mining::fp_growth::set_item |
( |
const std::string & |
_item | ) |
|
|
inline |
Set the item column name.
- Parameters
-
[in] | item | Name of the item column. |
Definition at line 80 of file fp_growth.hpp.
◆ set_options()
void turi::ml_model_base::set_options |
( |
const std::map< std::string, flexible_type > & |
_options | ) |
|
|
inherited |
Set one of the options in the algorithm.
The value are checked with the requirements given by the option instance.
- Parameters
-
[in] | name | Name of the option. |
[in] | value | Value for the option. |
◆ set_property()
variant_type turi::model_base::set_property |
( |
const std::string & |
property, |
|
|
variant_map_type |
argument |
|
) |
| |
|
inherited |
Sets a property. The new value of the property should appear in the argument map under the key "value".
◆ train()
void turi::pattern_mining::fp_growth::train |
( |
const gl_sframe & |
data | ) |
|
Train FP-Growth on the full dataset.
- Parameters
-
◆ uid()
virtual const std::string& turi::model_base::uid |
( |
| ) |
|
|
pure virtualinherited |
Returns a unique identifier for the toolkit class. It can be any unique ID. The UID is only used at runtime (to determine the concrete type of an arbitrary model_base instance) and is never stored.
Note: this function is typically overridden using the BEGIN_CLASS_MEMBER_REGISTRATION macro.
Implemented in turi::model_proxy.
◆ validate()
void turi::pattern_mining::fp_growth::validate |
( |
const gl_sframe & |
data, |
|
|
const std::string & |
item, |
|
|
const std::vector< std::string > & |
features |
|
) |
| const |
Validate the input data.
- Parameters
-
[in] | data | gl_sframe of data in stacked format. |
[in] | item | Name of the item column. |
[in] | features | Names of the feature columns. |
◆ state
std::map<std::string, variant_type> turi::ml_model_base::state |
|
protectedinherited |
The documentation for this class was generated from the following file: